Python Pandas – MultiIndex内のラベルまたはラベルタプルの位置を取得する方法

PandasでMultiIndex（マルチインデックス）内の特定のラベル、またはラベルのタプルが存在する位置を取得するには、MultiIndex.get_loc()メソッドを使用します。このメソッドは、指定したラベルに対応する整数位置やスライスオブジェクトを返します。

必要なライブラリのインポート

まず、必要なライブラリをインポートします。

import pandas as pd

MultiIndexの作成

MultiIndexとは、Pandasオブジェクトに対する多階層（階層的）なインデックスオブジェクトのことです。ここでは、配列からMultiIndexを作成します。

multiIndex = pd.MultiIndex.from_arrays([list('pqrrss'), list('stuvwx')])

MultiIndexの表示

作成したMultiIndexを表示して確認します。

print("The MultiIndex...\n", multiIndex)

ラベルの位置を取得する

get_loc()メソッドを使って、ラベル「s」の位置を取得します。第1レベルに「s」を持つエントリは複数あるため、この場合はスライスオブジェクトとして位置範囲が返されます。

print("\nGet the locations in MultiIndex...\n", multiIndex.get_loc('s'))

完全なコード例

以下に、ここまでの処理をまとめた完全なコードを示します。

import pandas as pd

# MultiIndexはPandasオブジェクトのための多階層（階層的）なインデックスオブジェクト
multiIndex = pd.MultiIndex.from_arrays([list('pqrrss'), list('stuvwx')])

# MultiIndexを表示
print("The MultiIndex...\n", multiIndex)

# MultiIndexのレベルを表示
print("\nThe levels in MultiIndex...\n", multiIndex.levels)

# ラベル「s」の位置を取得
print("\nGet the locations in MultiIndex...\n", multiIndex.get_loc('s'))

実行結果

上記のコードを実行すると、以下の出力が得られます。

The MultiIndex...
MultiIndex([('p', 's'),
            ('q', 't'),
            ('r', 'u'),
            ('r', 'v'),
            ('s', 'w'),
            ('s', 'x')],
           )

The levels in MultiIndex...
   [['p', 'q', 'r', 's'], ['s', 't', 'u', 'v', 'w', 'x']]

Get the locations in MultiIndex...
   slice(4, 6, None)

解説

出力結果からわかるように、ラベル「s」は第1レベルのインデックスとして位置4と5（タプル('s', 'w')と('s', 'x')）に存在します。そのため、get_loc()メソッドは単一の整数ではなく、slice(4, 6, None)というスライスオブジェクトを返します。

もし単一のラベルタプル（例：('s', 'w')）を指定した場合、戻り値は該当する1つの整数位置となります。このように、get_loc()は検索対象が一意かどうかによって、整数・スライス・ブール配列のいずれかを柔軟に返す便利なメソッドです。