K = 4 a = np.array([0, 8, 0, 4, 5, 8, 8, 0, 4, 2]) # 最大的 k 個元素的索引 print(np.argpartition(a, -K)[-K:]) # [4 1 5 6] # 最小的 k 個元素的索引 print(np.argpartition(a, K)[:K]) # [7 0 2 9]

總結

以上是生活随笔為你收集整理的python - 输出最大/最小的 k 个元素的索引的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。