Python用於取top N的模塊——heapq

Python有一個專門用於取前幾大或前幾小的模塊，就是heapq，下面代碼說明：

一、對列表取top N

>>> import heapq
>>> list1=[2, 3, 3, 4, 7, 9]
>>> nLargestList=heapq.nlargest(3,list1)
>>> print nLargestList
[9, 7, 4]
>>> nSmallestList=heapq.nsmallest(3,list1)
>>> print nSmallestList
[2, 3, 3]
>>> 

nlargest用於取前n大，相反，nsmallest用於取前n小。特此記錄

二、對字典取top N

可以用類似列表的方式，把字典的key或value取top N，然後再做相應處理即可。