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]

 

posted @ 2018-03-07 10:21  模式识别与计算机视觉  阅读(638)  评论(0编辑  收藏  举报