pythonで配列中から大きい値順のインデックスを取得する

Question

pythonでは配列中の最大値の添字を
numpy.argmax などで得られますが

もしかしてpythonでは
最大値からn番目の値の添字を取得する
というような事をスマートに実行する方法があったりするでしょうか？

Comments

[jcs502ulf]

得たいのは「最大値からn番目の要素」ではなく「最大値からn番目の添字」で間違いないですか？

[tak__tak]

要素も得たいですが添字も得たいです。

[cateye]

えっと、[1,3,5,2,4,8,7,0]の場合、最大値は８で添字は５ですが２番目の添字と言うのは７(0)の事でしょうか、それとも昇順に見た場合の添字３(5)の事でしょうか？

[tak__tak]

1番目は　　8[5]
2番目は　　7[6]
3番目は　　5[2]
4番目は　　4[4]
5番目は　　3[1]
6番目は　　2[3]
7番目は　　1[0]
8番目は　　0[7]

Answer 1

最大値が複数ないことが前提ですが。

max([(v,i) for (i,v) in enumerate(arr)])[1]

なんかいろいろ書き方ありそうですね。

#####訂正

arr = [3,5,1,3,2]
n = 4
sorted( [(v,i) for (i,v) in enumerate(arr)] )[-n][1] #4

意外と難しかった……　もっとスマートなのありそうですねえ

Comments

[Lhankor_Mhy]

あ、勘違いでした。n番目かあ……

[Lhankor_Mhy]

訂正しました。

[tak__tak]

(v,i) for (i,v)は
どういう意味なんでしょうか

[Lhankor_Mhy]

> タプルやリスト間の比較では、対応する各要素の比較結果を使って辞書的な 比較が行われます。このため、二つのシーケンスを等価にするためには、各要素が 完全に等価でなくてはならず、シーケンスは同じ型で同じ長さをもっていなければ なりません。

> 二つのシーケンスが等価でない場合、異なる値を持つ最初の要素間での比較に 従った順序関係になります。例えば、cmp([1,2,x], [1,2,y]) は cmp(x,y) と等しい結果を返します。片方の要素に対応する要素が 他方にない場合、より短いシーケンスが前に並びます (例えば、 [1,2] < [1,2,3] となります)。
http://docs.python.jp/2.4/ref/comparisons.html

ということらしいので、比較したい要素を前に出しました。

Answer 2

• maxとargmax http://python.dogrow.net/?p=117

...
max ( n ) は、n次元目の最大値を取得できる。
argmax ( n ) は、n次元目の最大値を持つ要素のインデックスを取得できる。
...

Answer 3

要素は以下で取得できると思いますが、添字はlen(arr)を利用できませんか？

arr = ["a", "b", "c"]
arr[-1] # "c"
arr[-2] # "b"

Answer 4

arr.argsort()[n]