如何修改脚本用生成器高效获取任意可迭代对象的最后N项？

嘿，我完全懂你碰到的这个麻烦——生成器（还有像range()这种惰性可迭代对象）确实不支持下标操作，因为它们不会一次性把所有元素加载到内存里，而是按需生成，自然没法直接用[-n:]这种方式取最后几项。不过咱们可以用滑动窗口的思路来解决，既保证内存高效，又能准确拿到想要的结果。

最优解决方案：用collections.deque实现固定长度队列 #

Python标准库的collections.deque有个非常好用的maxlen参数，当你给它设置这个参数后，队列会自动维护固定长度——一旦新元素加入导致队列长度超过maxlen，最前面的元素就会被自动弹出。用这个特性来遍历可迭代对象，内存里始终只会保留最多N个元素，完美适配超大可迭代对象的场景。

代码实现非常简洁：

from collections import deque

def get_last_n(iterable, n):
    # 遍历可迭代对象，deque自动维护最后n个元素
    return list(deque(iterable, maxlen=n))

为什么这个方法可行？ #

• 内存效率：不管你的可迭代对象是range(10000000)还是一个超大的生成器，deque只会在内存中存储最近的N个元素，不会加载整个序列。
• 遍历一次即可：整个过程只需要遍历可迭代对象一次，时间复杂度是O(len(iterable))，这是最优的时间复杂度（毕竟你必须遍历到最后才能知道最后N项是什么）。

测试示例 #

比如你用超大的range测试：

result = get_last_n(range(10000000), 5)
print(result)  # 输出: [9999995, 9999996, 9999997, 9999998, 9999999]

这个过程中内存占用极低，不会因为range的大小而飙升。

为什么你之前的尝试报错？ #

如果之前你尝试了类似list((x for x in iterable))[-n:]的写法，虽然能得到结果，但会把整个可迭代对象转换成列表，完全失去了生成器的内存优势；而如果直接对生成器用[-n:]，就会触发TypeError: 'generator' object is not subscriptable，因为生成器本身没有实现下标访问的接口。

而用deque的方式，既避免了加载整个序列，又绕开了生成器不能下标的问题，是最适合这个场景的方案。

内容的提问来源于stack exchange，提问作者MarkS