python yield函数

python yield函数

简单地讲,yield 的作用就是把一个函数变成一个 generator,带有 yield 的函数不再是一个普通函数,Python 解释器会将其视为一个 generator,调用带有它的函数不会执行函数,而是返回一个 iterable 对象!在 for 循环执行时,每次循环都会执行 even_number 函数内部的代码,执行到 yield n 时,even_number函数就返回一个迭代值,下次迭代时,代码从 yield n 的下一条语句继续执行,而函数的本地变量看起来和上次中断执行前是完全一样的,于是函数继续执行,直到再次遇到 yield。

也可以手动调用 even_number(5) 的 next() 方法(因为even_number(5) 是一个 generator 对象,该对象具有 next() 方法),这样我们就可以更清楚地看到 fab 的执行流程:

for i in _num(10):
    print(i)

等价于:
i = _num(10)
print(i.__next__(),i.__next__()...)
>>>0 2

我们可以得出以下结论:

一个带有 yield 的函数就是一个 generator,它和普通函数不同,生成一个 generator 看起来像函数调用,但不会执行任何函数代码,直到对其调用 next()(在 for 循环中会自动调用 next())才开始执行。虽然执行流程仍按函数的流程执行,但每执行到一个 yield 语句就会中断,并返回一个迭代值,下次执行时从 yield 的下一个语句继续执行。看起来就好像一个函数在正常执行的过程中被 yield 中断了数次,每次中断都会通过 yield 返回当前的迭代值。

yield 的好处是显而易见的,把一个函数改写为一个 generator 就获得了迭代能力,比起用类的实例保存状态来计算下一个 next() 的值,不仅代码简洁,而且执行流程异常清晰。

如何判断一个函数是否是一个特殊的 generator 函数?可以利用 isgeneratorfunction 判断:

使用 isgeneratorfunction 判断函数是否是 generator 函数

>>>from inspect import isgeneratorfunction >>> isgeneratorfunction(fab) True