【Python・Pandas】MultiIndexから特定のレベルを削除したインデックスを返す方法

PandasのMultiIndex.droplevel()メソッドを使用すると、マルチインデックスから指定したレベルを削除した新しいインデックスを取得できます。まずは、必要なライブラリをインポートしましょう。

import pandas as pd

次に、マルチインデックスを作成します。namesパラメータを指定すると、インデックスの各レベルに名前を設定できます。

multiIndex = pd.MultiIndex.from_arrays([[5, 10], [15, 20], [25, 30], [35, 40]],
names=['a', 'b', 'c', 'd'])

レベルを削除するには、droplevel()メソッドの引数に、削除したいレベルの位置（0始まりの整数）を渡します。ここでは、位置「3」にあたる4番目のレベルを削除しています。

print("\nDropping a level...\n",multiIndex.droplevel(3))

サンプルコード

以下に完全なコード例を示します。

import pandas as pd

# マルチインデックスを作成
# namesパラメータで各レベルに名前を設定
multiIndex = pd.MultiIndex.from_arrays([[5, 10], [15, 20], [25, 30], [35, 40]],
names=['a', 'b', 'c', 'd'])

# マルチインデックスを表示
print("Multindex...\n",multiIndex)

# レベルを削除
# 削除するレベルの位置を引数として渡す
print("\nDropping a level...\n",multiIndex.droplevel(3))

実行結果

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

Multindex...
MultiIndex([( 5, 15, 25, 35),(10, 20, 30, 40)],names=['a', 'b', 'c', 'd'])

Dropping a level...
MultiIndex([( 5, 15, 25),(10, 20, 30)],names=['a', 'b', 'c'])

補足：レベル名を指定して削除することも可能

droplevel()メソッドには、レベルの位置だけでなく、namesパラメータで設定したレベル名を直接指定することもできます。例えば、上記の例と同じ結果を得るには、次のように書けます。

print(multiIndex.droplevel('d'))

このように、位置番号よりもレベル名で指定した方が、コードの意図が明確になり可読性が向上します。状況に応じて使い分けるとよいでしょう。