Python每日一练0004

问题

如何保存迭代对象的最后N个元素

例如保存列表['a', 'b', 'c', 'd']的最后2个元素

或者保存某个迭代器对象的最后5个元素

解决方案

对于列表、元组这样的数据结构，可以使用切片来很方便的实现，例如保存列表l的最后3个元素就可以直接l[-3:]

但有些时候，我们操作的可迭代对象不能使用切片

这个时候可以使用collections库里的deque（双端队列）数据结构

from collections import deque

def foo():
    for i in range(100):
        yield i
        
last_5 = list(deque(foo(), maxlen=5))
print(last_5)
复制代码

deque能做到这一点，是因为如果deque指定了maxlen，那么当deque的元素超过maxlen时，就会从另一端把端头的元素删掉，再插入新的元素

也就是说，当deque指定了maxlen，deque中的元素永远最多只有maxlen个

这里虽然也可以使用list来模拟出这种效果，但并不能使用list，原因是因为list如果删除头部的元素，时间复杂度会是O(n)，而deque删除头部和尾部元素的时间复杂度都是O(1)

讨论

deque是一个类似于list的数据结构，但又与list有许多不同之处，这里详细介绍一下

• 构造deque对象时，如果不指定maxlen，那么deque的元素个数会不断增加；如果指定了maxlen，那么当元素的个数超过maxlen时再插入，就会先删掉另一端队头的元素，然后再插入新的元素
• deque也支持下标访问，访问两端的队头元素时，时间复杂度为O(1)，但访问中间的元素时，时间复杂度会退化到O(n)。这一点与list不同，list随机访问任意一个位置的元素时间复杂度都是O(1)
• deque删除两端元素的复杂度都是O(1)；而list删除最后一个元素的复杂度是O(1)，删除第一个元素的复杂度是O(n)，这是因为后面的元素需要向前移位

来源

Python Cookbook

关注

欢迎关注我的微信公众号：python每日一练