任何可通过生成器迭代的最后 N 项

Question

我正在编写一个脚本，它将任何可迭代的最后 N 项作为列表返回。

以下是有效的：

a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
b = (n**2 for n in range(10))


def tail(iterable, n):
    x = list(iterable)
    if n < 0:
        return []
    else:
        return list(x[-n:])


print(tail(b, 3))
print(tail(a, 4))

我想做最后一个改变。如果迭代很大，比如 range(10000000)，我不想预先生成整个列表。

我想提高内存效率，即使用生成器。

我做了几次尝试，包括这个：

a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
b = (n**2 for n in range(10))


def tail(iterable, n):
    if n < 0:
        return []
    else:
        return list([(yield x) for x in iterable][-n:])


print(tail(b, 3))

但我每次都遇到同样的错误：
TypeError：“生成器”对象不可下标

如何修改我的脚本以迭代生成器而不是列表，以便生成相同的输出（必须是列表）而不创建大的 可预先迭代。

Answer 1

您可以为此使用deque 并将其maxlen 指定为n：

from collections import deque

def tail(iterable, n):
    if n < 0:
        return []
    else:
        return deque(iterable, n)

将return 包装在list 调用中，以便在需要时将结果作为列表返回。 deque 不会在迭代器之外创建中间列表，从而降低空间复杂度。