python - ソートされたリストからインデックスを抽出する

Question

ソートされていないリストがあり、 を使用してソートするとしnp.sortます。numpy を使用して元のリストからソートされたリストのインデックスを取得する方法はありますか?

Answer 1

最も簡単な方法は、配列に位置インデックスを追加してから、2 次元配列を並べ替えることです。これにより、並べ替えられたデータと元の位置インデックスの両方が同時に得られます。

（ソートされたデータではなく）インデックスのみが必要な場合は、argsortを使用します。

>>> from numpy import array
>>> arr = array([10, 5, 80, 20, 70, 18])
>>> arr.argsort()
array([1, 0, 5, 3, 4, 2])