python -- 生成器

生成器

创建一个generator，有很多种方法。第一种方法很简单，只要把一个列表生成式的[]改成()，就创建了一个generator

>>> L = [x * x for x in range(10)]
>>> L
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> g = (x * x for x in range(10))
>>> g
<generator object <genexpr> at 0x1022ef630>

要一个一个打印出来，可以通过next(g)函数。每执行一次就可以打印下一个数据。因为generator也是可迭代对象，也可以直接用for循环print。并不需要关心StopIteration的错误。（但是有时候需要最后一个返回值时需要）

另一个生成生成器的方法就是把函数定义里面的print变成yield,这样这个函数就成了生成器。再调用生成器时先把生成器赋值给一个变量，实际要print输出时把next(变量)给到变量进行输出即可。最后要捕获输出结束的一个错误值进行生成器的输出退出。

def fib(max):
    n,a,b=0,0,0
    while n<max:
        yield b
        a,b=b,a+b
        n=n+1
    return 'done'

>>> g = fib(6)
>>> while True:
...     try:
...         x = next(g)
...         print('g:', x)
...     except StopIteration as e:
...         print('Generator return value:', e.value)
...         break
...
g: 1
g: 1
g: 2
g: 3
g: 5
g: 8
Generator return value: done