yield_atom ::= "(" yield_expression ")"
yield_expression ::= "yield" [expression_list]
在 Python 中带有 yield 的函数被称之为 generator(生成器), 简单地讲,yield 的作用就是把一个函数变成一个 generator。我们用生成斐波那契数列说明:
>>> def fab(max): n, a, b = 0, 0, 1 while n < max: yield b a, b = b, a + b n = n + 1
执行
>>> for n in fab(5): print(n) 1 1 2 3 5
带有 yield 的函数不再是一个普通函数,Python 解释器会将其视为一个 generator,调用 fab(5) 不会执行 fab 函数,而是返回一个 iterable 对象!在 for 循环执行时,每次循环都会执行 fab 函数内部的代码,执行到 yield b 时,fab 函数就返回一个迭代值,下次迭代时,代码从 yield b 的下一条语句继续执行,而函数的本地变量看起来和上次中断执行前是完全一样的,于是函数继续执行,直到再次遇到 yield。看起来就好像一个函数在正常执行的过程中被 yield 中断了数次,每次中断都会通过 yield 返回当前的迭代值。
也可以手动调用 fab(5) 的 __next__() 方法(当generator使用__next__()方法时,返回当前迭代值),这样我们就可以更清楚地看到 fab 的执行流程:
>>> f=fab(5) >>> f.__next__() 1 >>> f.__next__() 1 >>> f.__next__() 2 >>> f.__next__() 3 >>> f.__next__() 5 >>> f.__next__() Traceback (most recent call last): File "", line 1, in f.__next__() StopIteration
最后一个之所以报错是因为没有return返回值。在一个生成器中,如果没有return,则默认执行到函数完毕;如果遇到return,如果在执行过程中 return,则直接抛出 StopIteration 终止迭代。
原文地址:http://ypyu1989.blog.51cto.com/6786671/1535564