Python获取列表的最大n个元素或最小n个元素

一日一技:在Python里面如何获取列表的最大n个元素或最小n个元素?

我们知道,在Python里面,可以使用 maxmin获得一个列表的最大、最小的元素:

a =  [4,  2,  -1,  8,  100,  -67,  25]
max_value = max(a)
min_value = min(a)

print(max_value)

print(min_value)

运行效果如下图所示:

image

那么问题来了,如何获取最大的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和列表的长度相差无几,那么先排序再切片的性能会更高一些。

学习来源

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容