python - 在 Python 3 中使用生成器的类似方法

Question

我想了解 yield/generators 在 python 中的使用。我想知道，如果给定一个数字列表(例如:[1, 2, 3, 4, 5, 6, 7, 8, 9])，以下两个语句是等效的。如果不是，有什么区别？

x = (n for n in numbers)
return x


for n in numbers:
    yield n

Answer 1

据我所知，它们几乎等同于 PEP 289 中指定的:

g = (x**2 for x in range(10))
print g.next()

is equivalent to:

def __gen(exp):
    for x in exp:
        yield x**2
g = __gen(iter(range(10)))
print g.next()

然而，它只是几乎等价，在你的情况下的区别是何时 iter(numbers) 被评估:

>>> def f1(l):
...     x = (n for n in l)
...     return x
... 
>>> def f2(l):
...     for n in l:
...         yield n
...         

>>> a1 = f1(1)
TypeError: 'int' object is not iterable

>>> a2 = f2(2)  # works
>>> next(a2)    # fails only after calling "next"
TypeError: 'int' object is not iterable

f1 将立即尝试执行 iter(l) 而 f2 将仅执行 iter(l)在其上调用 next 之后。

要真正等效，您需要将 iter(numbers) 传递给使用 yield 的函数。