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