找出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]
标签:blog nbsp top element int index class python 索引
找出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]
python list sorting, return top N large elements index
标签:blog nbsp top element int index class python 索引
原文地址:https://www.cnblogs.com/changkaizhao/p/8520844.html