脚本宝典收集整理的这篇文章主要介绍了Python实用技法第3篇:找到最大或最小的N个元素,脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。
上一篇文章:Python实用技法第2篇:使用deque保留最新的N个元素
下一篇文章:Python实用技法第4篇:实现优先级队列
1、需求????
我们想在某个集合中找出最大或最小的N个元素
2、解决方案????
heapq模块中有两个函数:nlargest()和nsmallest()
代码:
import heapq
nums=[1,444,66,77,34,67,2,6,8,2,4,9,556]
PRint(heapq.nlargest(3,nums))
print(heapq.nsmallest(3,nums))
结果:
[556, 444, 77]
[1, 2, 2]
这个两个函数都可以接受一个参数key,从而允许他们可以工作在更加复杂的数据结构上:
代码:
import heapq
portfolio=[
{'name':'IBM','shares':100,'price':91.1},
{'name':'AAPL','shares':50,'price':543.22},
{'name':'FB','shares':200,'price':21.09},
{'name':'HPQ','shares':35,'price':31.75},
{'name':'YHOO','shares':45,'price':16.35},
]
cheap=heapq.nsmallest(3,portfolio,key=lambda s:s['price'])
expensive=heapq.nlargest(3,portfolio,key=lambda s:s['price'])
print(cheap)
print(exPEnsive)
结果:
[{'name': 'YHOO', 'shares': 45, 'price': 16.35}, {'name': 'FB', 'shares': 200, 'price': 21.09}, {'name': 'HPQ', 'shares': 35, 'price': 31.75}]
[{'name': 'AAPL', 'shares': 50, 'price': 543.22}, {'name': 'IBM', 'shares': 100, 'price': 91.1}, {'name': 'HPQ', 'shares': 35, 'price': 31.75}]
如果只是简单的查找最小或者最大的元素(N=1),那么使用Min()和max()会更快。
上一篇文章:Python实用技法第2篇:使用deque保留最新的N个元素
下一篇文章:Python实用技法第4篇:实现优先级队列
以上是脚本宝典为你收集整理的Python实用技法第3篇:找到最大或最小的N个元素全部内容,希望文章能够帮你解决Python实用技法第3篇:找到最大或最小的N个元素所遇到的问题。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。