Python Pandas – MultiIndexで指定ラベルの位置とスライスされたインデックスを取得する（レベルは削除しない）

はじめに

PandasのMultiIndex（マルチインデックス）において、指定したラベルやレベルの「位置」と「スライスされたインデックス」を同時に取得したい場合は、get_loc_level()メソッドを使用します。

通常、このメソッドは該当するレベルを結果から削除しますが、drop_levelパラメータをFalseに設定することで、レベルを保持したまま結果を得ることができます。

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

まず、Pandasをインポートします。

import pandas as pd

MultiIndexの作成

MultiIndexとは、pandasオブジェクトのための多階層（階層的）なインデックスオブジェクトです。ここでは、from_arrays()メソッドを使って2つの配列からMultiIndexを作成します。

multiIndex = pd.MultiIndex.from_arrays([list('pqrrss'), list('strvwx')],names=['One', 'Two'])

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

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

位置とスライスされたインデックスの取得

get_loc_level()メソッドにラベル「r」を渡し、位置とスライスされたインデックスを取得します。
レベルを削除せずに保持するため、「drop_level=False」を指定している点に注目してください。

print("\nGet the location and sliced index (avoid dropping the level)...\n",multiIndex.get_loc_level('r', drop_level=False))

コード例全体

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

import pandas as pd

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

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

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

# 位置とスライスされたインデックスを取得
# レベルを削除しないため、drop_levelパラメータにFalseを指定
print("\nGet the location and sliced index (avoid dropping the level)...\n",multiIndex.get_loc_level('r', drop_level=False))

実行結果

上記のコードを実行すると、次のような出力が得られます。

The MultiIndex...
MultiIndex([('p', 's'),
            ('q', 't'),
            ('r', 'r'),
            ('r', 'v'),
            ('s', 'w'),
            ('s', 'x')],
           names=['One', 'Two'])

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

Get the location and sliced index (avoid dropping the level)...
(slice(2, 4, None), MultiIndex([('r', 'r'),('r', 'v')],names=['One', 'Two']))

まとめ

get_loc_level()メソッドを使うことで、MultiIndex内の特定ラベルの位置情報とスライスされたインデックスを一度に取得できます。デフォルトでは該当レベルが削除されますが、drop_level=Falseを指定すれば、元の階層構造を保ったまま結果を受け取れるため、後続のデータ操作でレベル情報が必要な場合に便利です。