首页 > 代码库 > python生成器

python生成器

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 终止迭代。