Python Pandas – index.take()でインデックス位置を指定して新しいインデックスを返す方法

Pandasのindex.take()メソッドとは

Pandasで、指定したインデックス位置にある値だけを抜き出し、その新しいIndexオブジェクトを返したい場合は、index.take()メソッドを使用します。

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

import pandas as pd

Pandasインデックスの作成

次に、商品名を要素とするPandasインデックスを作成します。

index = pd.Index(['Electronics','Accessories','Decor', 'Books', 'Toys'], name ='Products')

Pandasインデックスの表示

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

指定した位置の値から新しいインデックスを取得

take()メソッドに位置番号のリスト（例：[1, 2]）を渡すことで、該当する値のみを含む新しいインデックスが返されます。

print("\nA new Index of the values selected by the indices...\n",index.take([1,2]))

サンプルコード

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

import pandas as pd

# Pandasインデックスの作成
index = pd.Index(['Electronics','Accessories','Decor', 'Books', 'Toys'], name ='Products')

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

# インデックス内の要素数を返す
print("\nNumber of elements in the index...\n",index.size)

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

# 指定したインデックス位置の値から新しいインデックスを取得
print("\nA new Index of the values selected by the indices...\n",index.take([1,2]))

出力結果

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

Pandas Index...
Index(['Electronics', 'Accessories', 'Decor', 'Books', 'Toys'], dtype='object', name='Products')

Number of elements in the index...
5

The dtype object...
object

A new Index of the values selected by the indices...
Index(['Accessories', 'Decor'], dtype='object', name='Products')

まとめ

このように、index.take()メソッドを使えば、元のインデックスから任意の位置の値を選択し、名前やdtypeなどの属性を保持したまま新しいIndexオブジェクトを簡単に生成できます。特定の行やラベルを抽出したいデータ処理の場面で非常に便利です。