Q&A
katoy様、アドバイスありがとうございます。ランク付けであればsicpyのrankdata関数のほうが理解しやすいです。こちらも試してみます。
pythonで乱数を生成した場合の最大値や２番目に大きい値などのインデックス番号を知るプログラムを作成したく思っております。
python
1 2コード 3import numpy as np 4a=np.random.randn(5) 5print(a) 6#aの乱数が 7[ 1.38270827 0.53661071 0.66367952 1.42236519 -0.5833445 ] 8a_max=max(enumerate(a)) 9#a_maxは(4, -0.5833444967072594)
調べていまして、enumerate関数を使用すればいいと思うのですが、上記の方法だとインデックス番号と最大値が取得できませんでした。
指定した数の順位などが知りたい場合に良い関数がありますでしょうか。
回答3件
0
既に解決済みですが、argsortを使うのも一つの手だと思います。
python
1import numpy as np 2 3array = np.array([1, 3, 2, 6, 5, 4]) 4print(np.argsort(array)) #[0 2 1 5 4 3]
投稿2018/09/15 05:02
総合スコア185
0
list のランキングをつける問題としてとらえてみます。
[10,12,11] に対して ランキング結果は [1, 2, 3] となります。
ランキング結果から、質問にあるようなインデックス値をえることは簡単にできます。
python3
1from scipy.stats import rankdata 2import numpy as np 3 4a = np.array([1, 3, 5, 2]) 5print(a) 6print("ranking: ", rankdata(a)) 7 8a = np.array([]) 9print(a) 10print("ranking: ", rankdata(a)) 11 12a = np.array([2, 2, 2, 1, 1, 1]) 13print(a) 14print("ranking: ", rankdata(a))
実行例
参考情報
- Python3で配列の要素についてランキングを作って云々する
- Pythonで順位をつける
投稿2018/09/15 04:53
総合スコア22324
0
ベストアンサー
例えば、小さい順にインデックス付きでソートするなら。
Python
1>>> print(a) 2[ 1.70061941 0.24255283 0.75208434 -0.06336105 -1.026424 ] 3>>> 4>>> from operator import itemgetter 5>>> sorted(enumerate(a), key=itemgetter(1)) 6[(4, -1.0264239982275112), (3, -0.063361054420085827), (1, 0.24255283127124649), (2, 0.75208434168339222), (0, 1.7006194092090856)]
最大／最小に関しては、np.argmax／np.argminが便利でしょう。
投稿2018/09/15 04:17
総合スコア35660
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2018/09/15 05:23