码迷,mamicode.com
首页 > 编程语言 > 详细

python list sorting, return top N large elements index

时间:2018-03-07 11:36:26      阅读:176      评论:0      收藏:0      [点我收藏+]

标签: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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!