ddda的CSDN

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

这里要借助到python的内置模块heapq，其原理是基于堆的，也就是二叉树import heapqa=[1,2,3,4,5]re1 = map(a.index, heapq.nlargest(3, a)) #求最大的三个索引 nsmallest与nlargest相反，求最小re2 = heapq.nlargest(3, a) #求最大的三个元素print(list(re1)...