欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python -heapq解决实际问题

程序员文章站 2022-07-12 17:22:56
...

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也可以实现优先级队列,根据优先级对元素进行排序。