pythonの配列においてn番目に小さい量が配列のどこに格納されているのかを知る方法はあるのか。

Question

pythonで配列を並び替える際には、sortedを用いると思います。例えば、次のような配列Aを並び替えたいときは、

A = [32,43,24,15,65,96,42,64]
sorted(A)

とすれば、
[15,24,32,42,43,64,65,96]
という結果が戻ってきますね。並び替えたあとの2番目の値24が、並び替える前にAのどこにあったのかを知る方法を知りたいです。（この場合、答えは3番目)

要は、2番目（より一般にn番目）に小さい値が、配列のどこに収納されているかのを知る方法を知りたいのです。

Answer 1

enumerateとsortedを組み合わせて使うことで実現できます。

>>> lst = [32,43,24,15,65,96,42,64]
>>> lst
[32, 43, 24, 15, 65, 96, 42, 64]
>>> list(enumerate(lst))
[(0, 32), (1, 43), (2, 24), (3, 15), (4, 65), (5, 96), (6, 42), (7, 64)]
>>> sorted(enumerate(lst), key=lambda x: x[1])
[(3, 15), (2, 24), (0, 32), (6, 42), (1, 43), (7, 64), (4, 65), (5, 96)]

enumerateがリストの添字と要素のタプルのリストを返すものなので、それをkey=lambda x: x[1]でsortedしてやる（要素の値の大小を見てソート）ことで該当する添字もわかります。

Comments

[physics303]

ありがとうございます。
上でLouiS0616さんがやっていることと全く同じという理解で正しいでしょうか。

[hayataka2049]

そうですね。まったく同じロジックです。ただし中身のタプルの順番が逆なので、そこだけ気をつけてください（LouiS0616さんの方はzipの中身の順番を変えてkey=itemgetter(1)とすれば私とまったく同じ結果になるはずです）

[physics303]

ありがとうございました。

Answer 2

numpy.argsortは試されましたか？

Comments

[physics303]

試してないです。調べてみます。

Answer 3

from itertools import count
from operator import itemgetter

A = [32, 43, 24, 15, 65, 96, 42, 64]
B = sorted(zip(A, count()), key=itemgetter(0))

こんな感じでインデックスを混ぜてソートするのが一つの手です。

並び替えたあとの2番目の値24が、並び替える前にAのどこにあったのかを知る方法を知りたいです。

B[1][-1]で取得できますね。

Comments

[LouiS0616]

なぜenumerateより先にcountを使ってしまったのだ自分は...

[hayataka2049]

内包要ります？

[LouiS0616]

要らないですね。ダブルでチョンボでした。
ご指摘ありがとうございます。

[physics303]

ありがとうございます。
えっと、B[1][-1]はB[1][1]の間違いではありませんか？

[LouiS0616]

Pythonではマイナスのインデックスを指定することで、『最後から何番目』を表現できるのです。

[physics303]

おお、ありがとうございます。

Answer 4

pandasのDataFrameに変換すると自動的にindexを割り振ってくれます。indexは値を使って並び替えても当初の番号を保持するので、必要なときに取り出す、当初の順番に並べ替える、などができので便利かと思います