Python Cookbook 1.4 找到最大或最小的 N 个元素

当我们需要在一个集合，或者列表中找到最大或最小的 N 个元素，这里我们用到的是 heapq 函数。

1、列表
heapq 模块中有两个函数 nlargest 和 nsmallest，可以接受一个数字 n 和列表作为参数，表示取其中最大或最小的 n 个数字。

返回的是一个已经排好序的列表，包含最大或者最小的几个元素。

2、字典列表
heapq 还可以通过添加 key 参数作用于字典等更复杂的数据结构。

3、利用 heapq 找出列表中最小的元素
heapq 中有一个函数 heapq.heapify() 可以对列表进行排序
然后利用 heapq.heapop() 函数每次 pop 出来的都是最小的元素：

注意

如果要查找的元素数量和元素总数相比，相对较小，那么可以直接使用 nlargest 和 nsmallest 函数。
如果只是想找一个元素，那么 max() 和 min() 函数则比较适合。
如果要找的元素个数和集合本身差不多大，更快速的方法是先对集合排序，然后做切片操作，比如：

sorted(nums)[:3]
sorted(nums)[-2:]