问题
在工作中,我们经常需要对列表字典集合等结构中的数据进行排序,取出最大或最小的N个值
解决方案:
这时就可以使用python中的heapq模块来快速实现我们的需求,在heapq模块中,有两个专门用来取最大和最小的N个元素的方法,分别是heapq.nlargest()和heapq.nsmallest()
语法:
nlargest(n, iterable, key=None)
nsmallest(n, iterable, key=None)
这两个方法都有三个参数:
参数 n :代表取前N个元素
参数 iterable :是你需要进行排序取值的可迭代对象
参数 key :这个参数可接受一个函数对象作为实参,当在对字典进行排序取值时这个参数非常有用,它可指定按字典中的Value进行比较排序
示例1:
假设我们要从一个列表中取出最大的5个数字,我们就可以这样写:
import heapq
numbers = [1, 3, 5, 7, 9, 10, 8, 6, 4, 2]
largest5 = heapq.nlargest(5, numbers)
print(largest5) # [10, 9, 8, 7, 6]
当我们要取的是最小的5个值时,我们可以这样写:
import heapq
numbers = [1, 3, 5, 7, 9, 10, 8, 6, 4, 2]
smallest5 = heapq.nsmallest(5, numbers)