生成器:(generator)
定义:在Python中,这种一边循环一边计算的机制,称为生成器。
意义:假如说创建一个100万个元素的list将会占用极大的内存,生成器可以不创建完整的list,而是可以不断推算出后续的元素,从而节省大量的空间。
列表生成式(列表):
>>>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 at 0x1022ef630>
L和g的区别:创建L和g的区别仅在于最外层的[]和(),L是一个list,而g是一个generator。
generator保存的是一种算法,是一个对象
如何遍历g的元素?
<1>next()函数,next(g),每次打印出一个元素,依次递推
<2>for语句来迭代其中的元素
斐波拉契数列(Fibonacci):
定义:除第一个和第二个数外,任意一个数都可由前两个数相加得到:
1, 1, 2, 3, 5, 8, 13, 21, 34, ...
斐波拉契数列用列表生成式写不出来,但是,用函数把它打印出来却很容易:
def fib(max):
n, a, b = 0, 0, 1
while n < max:
print(b) #将print(b)改为yield b,就是一个生成器(generator)
a, b = b, a + b #注意赋值语句
n = n + 1
return 'done'
输出结果:
>>> fib(6)
1
1
2
3
5
8
'done'
如果一个函数定义中包含yield关键字,那么这个函数就不再是一个普通函数,而是一个generator
函数是顺序执行,遇到return语句或者最后一行函数语句就返回。
而变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。(重要)