More than 5 years have passed since last update.

numpyで配列の要素とインデックスを降順に取り出す

Posted at 2016-02-22

numpyで配列の要素とインデックスを降順に取り出すには、sort関数とargsort関数を使います。

ただ、sort関数は昇順にソートし、argsort関数は昇順にソートしたインデックスの配列を返すので、[::-1]というスライスの書き方を使って逆順、つまり、降順にします。

import numpy as np

x = np.array([18, 7, 55, 31])

for i in range(len(x)):
    print np.argsort(x)[::-1][i], np.sort(x)[::-1][i]

次のように出力されます。