【部分引用】 argsort()_AILEARNER_L的博客-CSDN博客_argsort https://blog.csdn.net/AILEARNER_L/article/details/105509127

a=np.array([3,5,1,2,2,6]) a.argsort() #output:array([2, 3, 4, 0, 1, 5])

argsort返回數據從小到大的索引值

a=np.array([3,5,1,2,2,6]) a.argsort()[::-1] #output:array([5, 1, 0, 4, 3, 2])

這樣可以返回從大到小索引值

import numpy as np top_k=3 arr = np.array([91, 33, 92, 24, 25]) top_k_idx=arr.argsort()[::-1][0:top_k] print(top_k_idx) #[2 0 1]

這樣可以返回從大到小 的前三個值的索引值。“”

總結

以上是生活随笔為你收集整理的argsort（）返回数据从小到大的“索引值”的全部內容，希望文章能夠幫你解決所遇到的問題。

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