[python]リストの数値が小さい順にindexを表示

Question

リストの数値が小さい順にindexを表示したいです。例えば、次のようなリストがあって

lst=[9,3,6,1,2]

これを「数が大きい順に並べて、それぞれ元合った場所のindexを返す」ためにはどうすればよいでしょうか？

[3,4,1,2,0]
#最小値(1) のlst内でのindexは 3
#2つ目(2)　のlst内でのindexは 4
#最大値(9) のlst内でのindexは 0

Answer 1

numpyならargsortで一発です。

素のpythonでやるには、enumerateと組み合わせ、sortedのkey引数を活用してあげます。

>>> lst=[9,3,6,1,2]
>>> [i for i, _ in sorted(enumerate(lst), key=lambda x:x[1])]
[3, 4, 1, 2, 0]
>>> import numpy as np  # numpyの例
>>> np.array(lst).argsort()
array([3, 4, 1, 2, 0])

Answer 2

enumerateでインデックスを付けて丸ごとソートし、必要な部分だけ取り出せば良いです。

import operator

lst = [9, 3, 6, 1, 2]

dst = sorted(
    enumerate(lst), key=operator.itemgetter(1)
)
print(dst)

dst = [idx for idx, _ in dst]
print(dst)

実行結果 Wandbox

[(3, 1), (4, 2), (1, 3), (2, 6), (0, 9)]
[3, 4, 1, 2, 0]

Answer 3

sorted(range(len(lst)), key=lambda x: lst[x])

REPLでの評価

>>> lst=[9,3,6,1,2]

>>> sorted(range(len(lst)), key=lambda x: lst[x])
[3, 4, 1, 2, 0]

Answer 4

※コメントにてご指摘あったように、以下のコードにはいろいろ欠点があるのでおすすめしません。
hayataka2049氏のおっしゃるargsort関数が良いかと思います。

このようなコードはいかがでしょうか。

lst=[9,3,6,1,2]
lst2=sorted(lst)#小さい順にソート

#ソート後の要素がソート前どこにあったか探す
for i in range(len(lst2)):
    for j in range(len(lst)):
        if lst2[i]==lst[j]:#ソート後とソート前の要素が一致するとき
            lst2[i]=j#indexをlst2に与える
            break
        

ご参考になれば幸いです。

Comments

[hayataka2049]

それだと二重forで計算量が増える上、たとえばlst=[9,3,6,1,2,6]のようなケースに対応できないのでいまいちです。

[amahara_waya]

たしかにその通りです。ご指摘ありがとうございます。

Answer 5

提示結果からすると小さい順に並べてですよね。

lst=[9,3,6,1,2]
ret = list(map(lambda x:x[0], sorted(dict(enumerate(lst)).items(), key=lambda x:x[1])))
print(ret) # [3, 4, 1, 2, 0]

Comments

[hayataka2049]

辞書にした意義は

[can110]

あ…ないですね。