一日一技：在Python里面如何获取列表的最大n个元素或最小n个元素？
一日一技：在Python里面如何获取列表的最大n个元素或最小n个元素？
青南
发布于 2019-07-10 15:03:43
发布于 2019-07-10 15:03:43
我们知道，在Python里面，可以使用 max和 min获得一个列表的最大、最小的元素：
a = [4, 2, -1, 8, 100, -67, 25]max_value = max(a)min_value = min(a)
print(max_value)print(min_value)运行效果如下图所示：
那么问题来了，如何获取最大的3个元素和最小的5个元素？
你当然可以先排序，然后再取：
a = [4, 2, -1, 8, 100, -67, 25, 3, 4, 5, 6, 7, 55]a.sort()
print(f'最小的5个元素：{a[:5]}')print(f'最大的三个元素：{a[-3:]}')
那有没有其他办法呢？
答案是使用Python自带的 heapq：
import heapq
a = [4, 2, -1, 8, 100, -67, 25, 3, 4, 5, 6, 7, 55]max_three = heapq.nlargest(3, a)min_five = heapq.nsmallest(5, a)
print(f'最大的3个元素：{max_three}')print(f'最小的5个元素：{min_five}')运行效果如下图所示：
这里的 heapq是一个用于处理 堆这种数据结构的模块。它会把原来的列表转换成一个堆，然后取最大最小值。
需要注意，当你要取的是前n大或者前n小的数据时，如果n相对于列表的长度来说比较小，那么使用 heapq的性能会比较好。但是如果n和列表的长度相差无几，那么先排序再切片的性能会更高一些。
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2019-07-03，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读