python之生成器

1、列表推导式

可以通过列表推导式直接生成列表，如下所示：

>>> List=[x**2 for x in range(10)]
>>> List
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

虽然通过列表推导式可以快速生成一个列表，但是也有一定限制。例如，如果要生成一个包含100万个元素的列表，就需要很大的内存；如果我们仅仅需要访问前面几个元素，那后面绝大多数元素占用的空间都浪费了。

所以，如果列表元素可以按照某种算法推算出来，那我们是否可以在循环的过程中不断推算出后续的元素呢？这样就不必创建完整的list，从而节省大量的空间。在Python中，这种一边循环一边计算的机制，称为生成器（Generator）。

2、生成器

生成器保存的是算法，并且是可以迭代的，但是只可以读取它一次 ，因为它并不把所有的值放在内存中，它是实时地生成数据。

（1）生成器表达式

>>> generator=(x**2 for x in range(10))
>>> generator
<generator object <genexpr> at 0x10baefbe0>
>>> for g in generator:
...     print g
... 
0
1
4
9
16
25
36
49
64
81
>>> 

（2）yield关键字

如果一个函数定义中包含yield关键字，那么这个函数就不再是一个普通函数，而是一个generator，这就是定义generator的另一种方法。

生成器的编写方法和函数定义类似，只是在return的地方改为yield。生成器中可以有多个yield。当生成器遇到一个yield时，会暂停运行生成器，返回yield后面的值。当再次调用生成器的时候，会从刚才暂停的地方继续运行，直到下一个yield。生成器自身又构成一个循环器，每次循环使用一个yield返回的值。

#!/usr/bin/env python
# -*- coding:UTF-8 -*-

import sys

def fibonacci(Max):
	n,a,b = 0,0,1
	while n < Max:
		yield b
		a,b = b,a+b
		n=n+1

for g in fibonacci(int(sys.argv[1])):
	print g

$ python fibonacci.py 10
1
1
2
3
5
8
13
21
34
55

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