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

最新推荐文章于 2021-10-30 10:22:32 发布

weixin_44500320

最新推荐文章于 2021-10-30 10:22:32 发布

阅读量3.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： Python 文章标签：学习笔记

本文链接：https://blog.youkuaiyun.com/weixin_44500320/article/details/86470871

Python 专栏收录该内容

11 篇文章

订阅专栏

本文介绍在集合中寻找N个最大或最小元素的三种方法：使用max(), min(); 列表排序后切片; 使用heapq模块的nlargest(), nsmallest()函数。并详细说明了每种方法的最佳适用场景。

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

在集合中找出N个最大或者最小元素的方法，大体可以用一下几种：
1.用max(),min()
2.用列表排序而后切片，sorted[:N],sorted[-N:]
3.用heapq模块中的nlargest(),nsmallest()函数
那么以上几种方法分别什么时候使用呢，大致可以分一下几种情况：
1.当N=1的时候，使用max(),min()。例如在list中找出最大值和最小值：

max(list)
min(list)

2.当N远小于集合元素个数时：例如list (#len(list)=100) 中找出5个最大值和最小值，用heapq模块中的nlargest(),nsmallest()：

import heapq
heapq.nlargest(5,list)
heapq.nsmallest(5,list)

3.当N几乎和集合元素个数相等时，使用列表排序而后切片，sorted[:N],sorted[-N:]。例如在list (#len(list)=100) 中找出90个最大值和最小值:

sorted(list)[:90]
sorted(list)[-90:]

4.当集合的数据结构较为复杂时，也可使用heapq模块中的nlargest(),nsmallest()，这两个函数接受一个参数key,例如：

profiles = [
    {'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}
] 

cheap_3 = heapq.nsmallest(3,profiles,key=lambda s:s['price'])
expensive_3 = heapq.nlargest(3,profiles,key=lambda s:s['price'])