您的位置:首页 > 编程语言 > Python开发

[Python]heapq 堆模块

2014-05-03 16:15 246 查看
模块的名称是 heapq

如果一个比较大的列表,而且是要不断的更新数据,而你需要不停的从其中取出最大或者是最小的值。一种简单的方法就是每次,更新list之后就对其进行sort操作,但是如果list很大,这样每次排序不仅效率低下,资源消耗也相当的大。
使用场景: 优先级队列

这里就是对一个list 不断更新,得到最值

>>> import heapq
>>> a = []
>>> heapq.heappush(a, 4)
>>> heapq.heappush(a, 5)
>>> heapq.heappush(a, 1)
>>> heapq.heappush(a, 100)
>>> heapq.heappush(a, 33)
>>> heapq.heappop(a)
1
>>> heapq.heappush(a, 3)
>>> heapq.heappop(a)
3
>>> heapq.nlargest(1, a)
[100]
>>> heapq.nsmallest(1, a)
[4]
>>> heapq.heappush(a, 5)
>>> heapq.nsmallest(2, a)
[4, 5]
不管list怎么更新,总可以取出最值。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python heapd