儲存最後N個元素,找到最大或者最小的N個元素特殊佇列deque,heapq模組--cookbook讀書筆記
阿新 • • 發佈:2019-01-27
1. 特殊的佇列collections.deque,儲存最後N個元素
儲存有限的歷史記錄可算是collections.deque的完美應用場景了。
deque與list的區別是:
deque初始化的時候可以規定佇列大小,超過預定大小,新加入的元素會擠掉最早進入的元素;deque不僅僅支援append、pop實現後進先出,還支援appendleft、popleft對任何一邊元素進行操作;
>>> from collections import deque >>> q = deque(maxlen=3) >>> q.append(1) >>> q.append(2) >>> q.append(3) >>> q.append(4) >>> q deque([2, 3, 4], maxlen=3) >>> q.pop() 4 >>> q deque([2, 3], maxlen=3) >>> q.appendleft(1) >>> q deque([1, 2, 3], maxlen=3)
2. heapq 模組的函式應用於佇列(找到最大或者最小的N個元素)
我們想在某個集合中找出最大或者最小的N個元素,可以採用heapq模組中,有兩個函式------nlargest() 和nsmallest()
heapq提供一部分函式可以實現一些快捷的佇列應用:nlargest() 和nsmallest()獲得佇列中最大或最小的n個元素;
eapq.heapify(),heappush(),heappop()獲取最小元素;
>>> nums = [1, 8, 2, 23, 7, -4, 18, 23, 42, 37, 2] >>> import heapq >>> print(heapq.nlargest(3, nums)) [42, 37, 23]
import heapq protfolio=[ {'name':'IBM','shares':100,'price':91.1}, {'name':'FB','shares':200,'price':21.09}, {'name':'AAPL','shares':50,'price':543.22}, {'name':'HPQ','shares':35,'price':31.75}, {'name':'YHOO','shares':45,'price':16.35}, {'name':'ACME','shares':75,'price':115.65} ] cheap=heapq.nsmallest(3,protfolio,key=lambda s:s['price']) print(cheap) expensive=heapq.nlargest(3,protfolio,key=lambda s:s['price']) print(expensive)
特別指出,nlargest() 和nsmallest()支援第三個引數key=callable,這個key和sorted,max,min用法一樣