利用Python获取数组或列表中最大的N个数及其索引

shandler

于 2020-09-19 10:29:45 发布

阅读量2.3k

点赞数 1

原文链接：https://blog.youkuaiyun.com/qq_17249717/article/details/86242440

版权

本文介绍如何利用Python的heapq模块求解列表中的最大值及其索引，并通过实例展示了nlargest函数的用法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import heapq
 
a=[43,5,65,4,5,8,87]

re1 = heapq.nlargest(3, a) #求最大的三个元素，并排序
re2 = map(a.index, heapq.nlargest(3, a)) #求最大的三个索引    nsmallest与nlargest相反，求最小

print(re1)
print(list(re2)) #因为re2由map()生成的不是list，直接print不出来，添加list()就行了