Python学习十：生成器

什么是生成器呢？在生成列表元素时，通过循环不断推算出后续的元素这种机制就叫做生成器。

生成器的创建方式：
方式一：只要把一个列表生成式的[ ] 改为 （ ），就创建了一个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 0x108cfcaf0>

创建L 和 g 的区别仅在于最外层 的 [ ] 和 （ ），L是一个list ， 而g是一个generator.
可以直接打印出list的每一个元素，但我们怎么打印出generator的每一个元素呢？
可以使用 generator 提供的 next() 方法，但通常为了打印出所有的元素，使用for循环：

>>> for n in g :
...  print n
... 
0
1
4
9
16
25
36
49
64
81

第二种定义方式：
使用 yield的关键字，如果一个函数中又了yield的关键字，它就不再是一个单纯的函数了，而是一个generator了。

>>> def odd():
...     print 'step 1'
...     yield 1
...     print 'step 2'
...     yield 2
...     print 'step 3'
...     yield 5
... 

最难理解的是 generator 和函数的执行流程：
函数的执行顺序是，遇到return语句或者最后一行函数语句就返回。
而变成generator的函数，在每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次返回的yield语句处继续执行。

>>> o = odd()
>>> o.next()
step 1
1
>>> o.next()
step 2
2
>>> o.next()
step 3
5
>>> o.next()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

总之：
generator是非常强大的工具，在Python中，可以简单地把列表生成式改成generator，也可以通过函数实现复杂逻辑的generator。
要理解generator的工作原理，它是在for循环的过程中不断计算出下一个元素，并在适当的条件结束for循环。对于函数改成的generator来说，遇到return语句或者执行到函数体最后一行语句，就是结束generator的指令，for循环随之结束。