Python cookbook（數(shù)據(jù)結(jié)構(gòu)與算法）保存最后N個(gè)元素的方法

更新時(shí)間：2018年02月13日 11:07:15 作者：壟上行

這篇文章主要介紹了Python數(shù)據(jù)結(jié)構(gòu)與算法保存最后N個(gè)元素的方法,涉及Python基于迭代器與生成器實(shí)現(xiàn)歷史記錄功能的相關(guān)操作技巧,需要的朋友可以參考下

本文實(shí)例講述了Python保存最后N個(gè)元素的方法。分享給大家供大家參考，具體如下：

問題：希望在迭代或是其他形式的處理過程中對(duì)最后幾項(xiàng)記錄做一個(gè)有限的歷史記錄統(tǒng)計(jì)

解決方案：選擇collections.deque。

如下的代碼對(duì)一系列文本行做簡單的文本匹配操作，當(dāng)發(fā)現(xiàn)有匹配時(shí)就輸出當(dāng)前的匹配行以及最后檢查過的N行文本：

from collections import deque
def search(lines, pattern, history=5):
  previous_lines = deque(maxlen=history)
  for line in lines:
    if pattern in line:
      yield line, previous_lines
    previous_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):
      for pline in prevlines:
        print(pline, end='')
      print(line, end='')
      print('-'*20)

正如上面的代碼一樣，當(dāng)編寫搜索某項(xiàng)記錄的代碼時(shí)，通常會(huì)用到含有yield關(guān)鍵字的生成器函數(shù)，將處理搜索過程的代碼與使用搜索結(jié)果的代碼解耦開來。具體生成器可參考本站迭代器和生成器相關(guān)內(nèi)容。

deque(maxlen=N)創(chuàng)建一個(gè)固定長度的隊(duì)列，當(dāng)加入新元素而隊(duì)列已滿時(shí)會(huì)自動(dòng)移除最老的那條記錄：

>>> 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)
>>>

盡管可以在列表上手動(dòng)完成這樣的操作（append、del），但隊(duì)列的這種解決方案要優(yōu)雅得多，運(yùn)行速度也快得多。

如果不指定隊(duì)列長度，則得到一個(gè)無界限的隊(duì)列，可在兩端執(zhí)行添加和彈出操作：

>>> q=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
>>> q
deque([1, 2])
>>>

更多關(guān)于Python相關(guān)內(nèi)容感興趣的讀者可查看本站專題：《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python加密解密算法與技巧總結(jié)》、《Python編碼操作技巧總結(jié)》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》及《Python入門與進(jìn)階經(jīng)典教程》

希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。

您可能感興趣的文章: