我们知道,在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和列表的长度相差无几,那么先排序再切片的性能会更高一些。