当前位置:Gxlcms > Python > python中找到最大或最小的N个元素的实现代码

python中找到最大或最小的N个元素的实现代码

时间:2021-07-01 10:21:17 帮助过:26人阅读

本篇文章给大家带来的内容是关于python中找到最大或最小的N个元素的实现代码,有一定的参考价值,有需要的朋友可以参考一下,希望对你有所帮助。

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中找到最大或最小的N个元素的实现代码的详细内容,更多请关注Gxl网其它相关文章!

人气教程排行