找到最大或最小的N个元素

最新推荐文章于 2021-09-29 11:54:05 发布

tianv5

最新推荐文章于 2021-09-29 11:54:05 发布

阅读量231

点赞数

本文链接：https://blog.csdn.net/sunt2018/article/details/100124398

版权

在某个集合中找出最大或最小的N个元素

heapq 模块的 nsmallest 和 nlargest是我们所需要的。

如果只是找最大和最小 max min 更适合我们(更快)

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},
   {'name': 'ACME', 'shares': 75, 'price': 115.65}
]

# 可以接受key值
cheap = heapq.nsmallest(3, portfolio, key=lambda s: s['price'])
expensive = heapq.nlargest(3, portfolio, key=lambda s: s['price'])

print(cheap)
print(expensive)