Python cookbook（数据结构与算法）保存最后N个元素的方法

2020-01-04 15:50:53

字体：大中小

来源：转载

供稿：网友

本文实例讲述了Python保存最后N个元素的方法。分享给大家供大家参考，具体如下：

问题：希望在迭代或是其他形式的处理过程中对最后几项记录做一个有限的历史记录统计

解决方案：选择collections.deque。

如下的代码对一系列文本行做简单的文本匹配操作，当发现有匹配时就输出当前的匹配行以及最后检查过的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)

Python,cookbook,数据结构与算法,最后N个元素

正如上面的代码一样，当编写搜索某项记录的代码时，通常会用到含有yield关键字的生成器函数，将处理搜索过程的代码与使用搜索结果的代码解耦开来。具体生成器可参考本站迭代器和生成器相关内容。

deque(maxlen=N)创建一个固定长度的队列，当加入新元素而队列已满时会自动移除最老的那条记录：

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

尽管可以在列表上手动完成这样的操作（append、del），但队列的这种解决方案要优雅得多，运行速度也快得多。

如果不指定队列长度，则得到一个无界限的队列，可在两端执行添加和弹出操作：

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

希望本文所述对大家Python程序设计有所帮助。

注：相关教程知识阅读请移步到python教程频道。

上一篇：python3学习笔记之多进程分布式小例子

下一篇：Python cookbook（数据结构与算法）从任意长度的可迭代对象中分解元素操作示例