Python Pandas入門：指定したラベルを削除して新しいインデックスを作成する方法

Pandasで、既存のインデックスから指定したラベルを削除して新しいインデックスを作成するには、index.drop()メソッドを使用します。このメソッドの引数に、削除したいラベルのリストを渡すだけでOKです。

なお、drop()メソッドは元のインデックスを書き換えるのではなく、削除後の新しいインデックスを返す点に注意してください。

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

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

import pandas as pd

2. インデックスの作成

ここでは、乗り物の名前を要素とするインデックスを作成します。

index = pd.Index(['Car','Bike','Truck','Ship','Airplane'])

3. インデックスの表示

作成したインデックスを確認してみましょう。

print("Pandas Index...\n",index)

4. ラベルを削除して新しいインデックスを作成

削除したいラベルをリストとしてdrop()メソッドに渡します。ここでは「Bike」と「Ship」を削除してみます。

print("\nUpdated index after deleting labels...\n",index.drop(['Bike', 'Ship']))

完全なサンプルコード

以下が実行コード全体です。

import pandas as pd

# インデックスを作成
index = pd.Index(['Car','Bike','Truck','Ship','Airplane'])

# インデックスを表示
print("Pandas Index...\n",index)

# データのdtype（データ型）を返す
print("\nThe dtype object...\n",index.dtype)

# 基となるデータの形状（shape）をタプルで返す
print("\nA tuple of the shape of underlying data...\n",index.shape)

# 削除するラベルを含むリストを渡して新しいインデックスを取得
print("\nUpdated index after deleting labels...\n",index.drop(['Bike', 'Ship']))

実行結果

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

Pandas Index...
Index(['Car', 'Bike', 'Truck', 'Ship', 'Airplane'], dtype='object')

The dtype object...
object

A tuple of the shape of underlying data...
(5,)

Updated index after deleting labels...
Index(['Car', 'Truck', 'Airplane'], dtype='object')

「Bike」と「Ship」が取り除かれ、「Car」「Truck」「Airplane」だけを含む新しいインデックスが作成されていることがわかります。

補足：存在しないラベルを指定した場合

drop()メソッドに存在しないラベルを渡すと、デフォルトではKeyErrorが発生します。エラーを回避したい場合は、errors='ignore'オプションを指定しましょう。

# 存在しないラベルが含まれていてもエラーにならない
index.drop(['Bike', 'Ship'], errors='ignore')

このように、index.drop()メソッドを使えば、不要なラベルを簡単に除外した新しいインデックスを安全に作成できます。