223

次のコードを検討してください。

avgDists = np.array([1, 8, 6, 9, 4])
ids = avgDists.argsort()[:n]

これにより、n最小要素のインデックスが得られます。argsortこれを降順で使用して、n最高の要素のインデックスを取得することは可能ですか?

4

9 に答える 9

4

配列のコピーを作成し、各要素に -1 を掛けることができます。
効果として、最大の要素の前が最小になります。
コピーの n 個の最小要素のインデックスは、元の n 個の最大要素です。

于 2013-05-10T16:01:38.460 に答える