电子生医小白的博客

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

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 nptop_k=3arr = np.array([91, 33, 92, 24, 25])top_k_