python获取列表中最大N个数及其索引

求前N个最大(小)元素我们首先先到的方法一定是堆排序,python中提供了内置的heapq模块可以实现堆排序,对外接口直接调用即可。参考了一下萝卜吃土豆的代码。

heapq可以对字典进行排序,基于此,我们求最大N个数及其索引的代码如下:

import heapq

testList = [1,2,3,4,5]
tmp = zip(range(len(testList)), testList)
large5 = heapq.nlargest(5, tmp, key=lambda x:x[1])
print(large5)  #Index and value of the largest 5 elements

需要注意的是,堆排属于不稳定排序

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值