找出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 模式识别与计算机视觉
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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]
以上是关于python list sorting, return top N large elements index的主要内容,如果未能解决你的问题,请参考以下文章
python中list.sort()与sorted()的区别
python学习 -- operator.itemgetter(), list.sort/sorted 以及lambda函数