Find centralized, trusted content and collaborate around the technologies you use most.
Teams
Q&A for work
Connect and share knowledge within a single location that is structured and easy to search.
次のコードを検討してください。
avgDists = np.array([1, 8, 6, 9, 4]) ids = avgDists.argsort()[:n]
これにより、n最小要素のインデックスが得られます。argsortこれを降順で使用して、n最高の要素のインデックスを取得することは可能ですか?
n
argsort
配列のコピーを作成し、各要素に -1 を掛けることができます。 効果として、最大の要素の前が最小になります。 コピーの n 個の最小要素のインデックスは、元の n 個の最大要素です。