python list sorting, return top N large elements index
找出numpy数组中最大的N个数的索引:
def findTopNindex(arr,N): return np.argsort(a)[::-1][:N]
测试:
test = np.array([2,5,6,3,4,6,4,8,6,5]) print(findTopNindex(test,3))
>[7 8 5 2]
找出numpy数组中最大的N个数的索引:
def findTopNindex(arr,N): return np.argsort(a)[::-1][:N]
测试:
test = np.array([2,5,6,3,4,6,4,8,6,5]) print(findTopNindex(test,3))
>[7 8 5 2]