从一个集合中查找最大最小的N个元素——Python heapq 堆数据结构

从一个集合中查找最大最小的N个元素——Python heapq 堆数据结构
Top N问题在搜索引擎、推荐系统领域应用很广， 如果用我们较为常见的语言，如C、C++、Java等，代码量至少也得五行，但是用Python的话，只用一个函数就能搞定，只需引入heapq(堆队列)这个数据结构即可。今天偶然看到这个库，特意记下之。

先看一个例子：

复制代码
1 >>> import heapq
2 >>> nums = [1,8,2,23,7,-4,18,23,42,37,2]
3 >>> print heapq.nlargest(3, nums)
4 [42, 37, 23]
5 >>>
6 >>> print heapq.nsmallest(3, nums)
7 [-4, 1, 2]
复制代码
是不是很简洁？
我们具体来看一下具体的函数定义。heapq有很多函数，最为堆，队列，可想而知，也就是那些push，pop之类的操作，详细请看官方文档：https://docs.python.org/2/library/heapq.html，在这里，我们只看Top N的两个函数，其他函数在用到的时候查看文档就好了。

1）、heapq.nlargest(n, iterable[, key])

从迭代器对象iterable中返回前n个最大的元素列表，其中关键字参数key用于匹配是字典对象的iterable，用于更复杂的数据结构中。

2）、heapq.nsmallest(n, iterable[, key])

从迭代器对象itera