电子生医小白的博客

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