python中的生成器

1.简介

在Python中，生成器是一种边循环边计算的惰性机制，需要我们手动去提取这些元素。

2.示例

(表达式   循环体   条件语句)

注：列表生成式的[]改成()，就创建了一个generator.

>>> L1=[1, 2, 3, 4, 5, 6]
>>> g=(x*x for x in L1)
>>> g
<generator object <genexpr> at 0x0268E968>
>>> next(g)  #也可以使用g.next()
1
>>> next(g)
4

也可以直接使用循环(避免抛出错误)： 

>>> for n in g:
      print(n)

3.函数生成器

斐波拉契数列（Fibonacci），除第一个和第二个数外，任意一个数都可由前两个数相加得到：

1, 1, 2, 3, 5, 8, 13, 21, 34, ...

函数：

def fib(max):
    n, a, b = 0, 0, 1
    while n < max:
        print b
        a, b = b, a + b
        n = n + 1

计算结果：

>>> fib(4)
1
1
2
3

如何将上述函数编程生成器呢？方法如下：把print b改为yield b。

函数：

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

计算结果：

>>> f = fib(6)
>>> f
<generator object fib at 0x104feaaa0>
>>> for n in fib(6):
... print(n)
1
1
2
3
5
8