Python保存最后N个元素
生活随笔
收集整理的這篇文章主要介紹了
Python保存最后N个元素
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
提出問題:
有時候,我們希望能在迭代或者其他處理過程中,保留最后幾個符合要求的內容記錄。
解決思路:
保存最后N個元素或者有限歷史記錄,可以用到collections.deque方法。例如,在一個文件里面查找,符合要求的文字,當發現有成功匹配的就輸出當前的匹配行以及最后檢查過的N行文本。
from collections import deque #導入deque模塊def search(lines,pattern,history=5):previous_lines=deque(maxlen=history)for line in lines: #循環遍歷if pattern in line:yield line ,previous_lines #yield 生成器,暫時保存返回當前變量previous_lines.append(line)if __name__ == '__main__':with open('SearchPython.txt') as f: #打開本地文件for line ,prevlines in search(f,'python',5): #在此文件中搜索,規則是python ,次數是5次for pline in prevlines:print(pline,end='') #下面是格式輸出print(line,end='')print('-'*20)本地一個SearchPython.txt文檔
12pythonpython10pppython1wepythonqqpython12查看搜索后結果:
12python -------------------- 12python python10 -------------------- 12python python10 pppython1 -------------------- 12python python10 pppython1 wepython -------------------- 12python python10 pppython1 wepython qqpython12 --------------------深入討論:
上面代碼中的deque(maxken=N),其實是創建了一個固定長度的隊列,有記錄來臨時會在尾部插入,當隊列滿時,自動移除最老的那條記錄。比如:
更簡單而又普遍的方式是,如果我們不指定隊列的大小,就會得到一個沒有邊界的隊列,可以在兩端執行添加和彈出操作,比如:
時間復雜度分析:
從隊列兩端添加或彈出元素的復雜度都是O(1),這一點是和列表是有區別的。當從列表的頭部插入或移除元素時,由于所有的元素都自動向前或向后移位,列表的時間復雜度為O(N)。
總結
以上是生活随笔為你收集整理的Python保存最后N个元素的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Python将序列分解为单独的变量
- 下一篇: Python从N个数中找到最大的K个数