本文實(shí)例講述了Python保存最后N個(gè)元素的方法。分享給大家供大家參考,具體如下:
問(wèn)題:希望在迭代或是其他形式的處理過(guò)程中對(duì)最后幾項(xiàng)記錄做一個(gè)有限的歷史記錄統(tǒng)計(jì)
解決方案:選擇collections.deque。
如下的代碼對(duì)一系列文本行做簡(jiǎn)單的文本匹配操作,當(dāng)發(fā)現(xiàn)有匹配時(shí)就輸出當(dāng)前的匹配行以及最后檢查過(guò)的N行文本:
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 previous_lines.append(line)# Example use on a fileif __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ù),將處理搜索過(guò)程的代碼與使用搜索結(jié)果的代碼解耦開來(lái)。具體生成器可參考本站迭代器和生成器相關(guān)內(nèi)容。
deque(maxlen=N)創(chuàng)建一個(gè)固定長(zhǎng)度的隊(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)>>> qdeque([1, 2, 3], maxlen=3)>>> q.append(4)>>> qdeque([2, 3, 4], maxlen=3)>>> q.append(5)>>> qdeque([3, 4, 5], maxlen=3)>>>
盡管可以在列表上手動(dòng)完成這樣的操作(append、del),但隊(duì)列的這種解決方案要優(yōu)雅得多,運(yùn)行速度也快得多。
如果不指定隊(duì)列長(zhǎng)度,則得到一個(gè)無(wú)界限的隊(duì)列,可在兩端執(zhí)行添加和彈出操作:
>>> q=deque()>>> qdeque([])>>> q.append(1)>>> q.append(2)>>> q.append(3)>>> qdeque([1, 2, 3])>>> q.appendleft(4)>>> qdeque([4, 1, 2, 3])>>> q.pop()3>>> qdeque([4, 1, 2])>>> q.popleft()4>>> qdeque([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ì)有所幫助。
新聞熱點(diǎn)
疑難解答
圖片精選