argsort（）返回数据从小到大的“索引值”

【部分引用】 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]

这样可以返回从大到小 的前三个值的索引值。“”