python获取数组或列表中最大的N个数及索引

这里要借助到python的内置模块heapq，其原理是基于堆的，也就是二叉树

import heapq

a=[1,2,3,4,5]
re1 = map(a.index, heapq.nlargest(3, a)) #求最大的三个索引    nsmallest与nlargest相反，求最小
re2 = heapq.nlargest(3, a) #求最大的三个元素
print(list(re1)) #因为re1由map()生成的不是list，直接print不出来，添加list()就行了
print(re2) 

输出结果为

[4, 3, 2]
[5, 4, 3]