【學習筆記】heapq_API

Python內置的heapq模塊

import heapq

1. heapq.heappush(heap,item) # 將 item 壓入到 heap 裏面

items = [1,2,9,7,3]
heapq.heappush(items,10)
items
[1, 2, 9, 7, 3, 10]

2. heapq.heappop(heap) # 將 heap 裏面的最小值 pop 出 heap ,heap 爲空的時候報 indexError 的錯誤

items
[1, 2, 9, 7, 3, 10]
heapq.heappop(items)
1
items
[2, 3, 9, 7, 10]

3. heapq.heappushpop(heap,item) # pop出 heap 中最小的元素,推入到 item

items
[2, 3, 9, 7, 10]
heapq.heappushpop(items,11)
2
items
[3, 7, 9, 11, 10]

4. heapq.heapify(x) # 將 list X 轉化爲 heap

nums = [1,10,9,8]
nums
[1, 10, 9, 8]
heap = list(nums)
heapq.heapify(heap)
heap
[1, 8, 9, 10]

5. heapq.heapreplace(heap,item):pop出最小值,推入item,heap的size不變

heap
[1, 8, 9, 10]
heapq.heapreplace(heap,100)
1
heap
[8, 10, 9, 100]

*6. heapq.merge(iterable):將多個可迭代合併,並且排好序,返回一個iterator

heap
[8, 10, 9, 100]
heap1 = [10,67,56,80,79]
h = heapq.merge(heap,heap1)
list(h)
[8, 10, 9, 10, 67, 56, 80, 79, 100]

P.S:
需要 說明的是這裏所謂的排序不是完全排序,只是兩個list對應位置比較,將小的值先push,然後大的值再與另外一個list的下一個值比較

7. heapq.nlargest(n,iterable,key):返回item中大到小順序的前N個元素,key默認爲空,可以用來指定規則如:function等來處理特定的排序

8. heapq.nsmallest(n,iterable,key):返回item中小到大順序的前N個元素,key默認爲空,可以用來指定規則如:function等來處理特定的排序

To create a heap, use a list initialized to[], or you can transform a populated list into a heap via functionheapify().

創建heap可以通過創建list,和使用heapify方法來實現

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章