Python Pandas入門：argsort()でインデックスをソートする整数インデックスを取得する方法

Pandasでインデックスを並べ替える（ソートする）ための整数インデックスを取得したい場合は、index.argsort()メソッドを使用します。このメソッドは、NumPyのargsortと同様に、インデックスの要素を昇順に並べた場合の元の位置（インデックス番号）を配列として返します。

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

import pandas as pd

Pandasインデックスの作成

次に、商品名を含むPandasインデックスを作成します。

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

インデックスの表示

作成したPandasインデックスを表示して確認します。

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

ソート用の整数インデックスを取得

index.argsort()を呼び出すことで、インデックスをソートする際に必要となる整数インデックスを取得できます。

res = index.argsort()

完全なサンプルコード

以下に、これまでの手順をまとめたコード例を示します。あわせて、要素数やデータ型の確認、argsort()の結果を使った実際のソート方法も紹介します。

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)

res = index.argsort()

# インデックスをソートする整数インデックスを返す
print("\nThe integer indices to sort the index...\n",res)

# 取得したインデックスを使って実際に並べ替えた結果を表示
print("\nOrdered..\n",index[res])

実行結果

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

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

Number of elements in the index...
5

The dtype object...
object

The integer indices to sort the index...
[1 3 2 0 4]

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

解説

出力結果を見ると、argsort()は[1 3 2 0 4]という整数インデックスの配列を返しています。これは「元のインデックスの1番目（Accessories）、3番目（Books）、2番目（Decor）、0番目（Electronics）、4番目（Toys）の順に並べ替えると昇順になる」という意味です。

さらに、この結果をindex[res]のように使うことで、実際にアルファベット順にソートされた新しいインデックスを簡単に取得できます。argsort()はデータフレームの行を特定の順序で並べ替えたい場合などにも応用できる便利なメソッドです。