Python Pandas – MultiIndexでラベルのシーケンスの位置を取得する方法

PandasのMultiIndexにおいて、ラベルのシーケンスに対応する位置（インデックス番号）を取得するには、MultiIndex.get_locs()メソッドを使用します。このメソッドは、指定したラベルがMultiIndex内のどの位置にあるかを特定したい場合に非常に便利です。

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

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

import pandas as pd

MultiIndexの作成

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

multiIndex = pd.MultiIndex.from_arrays([list('pqrrst'), list('kytssp')])

MultiIndexの表示

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

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

ラベルの位置を取得

get_locs()メソッドを使って、指定したラベルの位置を取得します。以下の例では、ラベル「's'」の位置を取得しています。

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

完全なコード例

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

import pandas as pd

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

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

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

# ラベルのシーケンスの位置を取得
print("\nGet the locations in MultiIndex...\n", multiIndex.get_locs('s'))

実行結果

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

The MultiIndex...
MultiIndex([('p', 'k'),
            ('q', 'y'),
            ('r', 't'),
            ('r', 's'),
            ('s', 's'),
            ('t', 'p')],
           )

The levels in MultiIndex...
   [['p', 'q', 'r', 's', 't'], ['k', 'p', 's', 't', 'y']]

Get the locations in MultiIndex...
   [4]

解説

出力結果からわかるように、ラベル「's'」はMultiIndexの4番目の位置（0始まり）に存在することが確認できます。get_locs()メソッドは、単一のラベルだけでなく、スライスやブール配列なども受け付けるため、柔軟な位置検索が可能です。階層的なデータ構造を扱う際に、特定のラベルの位置を素早く特定できる強力なメソッドです。