保留最近N個元素

問題

在迭代操作或者其他操作的時候，怎樣只保留最后有限幾個元素的歷史記錄？

解法

保留有限歷史記錄正是collections.deque大顯身手的時候。比如，下面的代碼在多行上面做簡單的文本匹配，并返回匹配所在行的前N行：

deque（double-ended queue，雙端隊列）

from collections import dequedef search(lines, pattern, history=5):previous_lines = deque(maxlen=history)for line in lines:if pattern in line:yield line, previous_lines# 將yield看成多次返回的returnprevious_lines.append(line)# Example use on a file if __name__ == '__main__':with open('somefile.txt') as f:for line, prevlines in search(f, 'python', 5):#迭代yield產生的內容for pline in prevlines:print(pline, end='')print(line, end='')print('-'*20)

假設somefile.txt內容為：

python 1 python 2 abc python 3 abc python 4 abc python 5 abc abc abc python 6 abc abc abc python 7 abc abc abc python 8 abc abc python 9 abc

運行上述程序得：

python 1 -------------------- python 1 python 2 -------------------- python 1 python 2 abc python 3 -------------------- python 1 python 2 abc python 3 abc python 4 -------------------- abc python 3 abc python 4 abc python 5 -------------------- abc python 5 abc abc abc python 6 -------------------- abc python 6 abc abc abc python 7 -------------------- abc python 7 abc abc abc python 8 -------------------- abc abc python 8 abc abc python 9 --------------------

討論

使用 deque(maxlen=N)構造函數會新建一個固定大小的隊列。當新的元素加入并且這個隊列已滿的時候，最老的元素會自動被移除掉。

>>> from collections import deque >>> q = deque(maxlen=3) >>> q.append(1) >>> q.append(2) >>> q.append(3) >>> q deque([1, 2, 3], maxlen=3) >>> q.append(4) >>> q deque([2, 3, 4], maxlen=3) >>> q.append(5) >>> q deque([3, 4, 5], maxlen=3)

---

盡管你也可以手動在一個列表上實現這一的操作 (比如增加、刪除等等)。但是這里的隊列方案會更加優雅并且運行得更快些。

其實，deque 類可以被用在任何你只需要一個簡單隊列數據結構的場合。如果你不設置最大隊列大小，那么就會得到一個無限大小隊列，你可以在隊列的兩端執行添加和彈出元素的操作。

>>> from collections import deque >>> q = deque() >>> q.append(1) >>> q.append(2) >>> q.append(3) >>> q deque([1, 2, 3]) >>> q.appendleft(4) >>> q deque([4, 1, 2, 3]) >>> q.pop() 3 >>> q deque([4, 1, 2]) >>> q.popleft() 4

在隊列兩端插入或刪除元素時間復雜度都是O(1)，而在列表的開頭插入或刪除元素的時間復雜度為O(N)。

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（1.3）：保留最近N个元素的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。