heapq模块实现了堆的排序算法,堆是二叉树,最大堆中父节点大于或等于两个子节点,最小堆父节点小于或等于两个子节点。
heapq中的两个函数nlargest()和nsmallest()函数可以找到一个序列中最大和最小的N个元素。
alists=[{"name":"xiaoming","age":18,"grade":90},{"name":"loihua","age":20,"grade":88.99},
{"name":"lili","age":22,"grade":98},
{"name":"wanger","age":20,"grade":78},
{"name":"wangsan","age":18,"grade":69},{"name":"maoyi","age":22,"grade":95}]
height=heapq.nlargest(2,alists,lambda s: s["grade"])
low=heapq.nsmallest(3,alists,lambda s: s["grade"])
heapq有两种方式创建堆, 一种是使用一个空列表,然后使用heapq.heappush()函数把值加入堆中,另外一种就是使用heap.heapify(list)转换列表成为堆结构,堆的一个重要特性就是heapq【0】总是最小的那个元素,所以可以通过heapq.heappop()方法找到堆中最小的元素。
使用heapq也可以实现优先级队列,根据优先级对元素进行排序。