生成器

生成器:(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语句处继续执行。(重要)

你可能感兴趣的:(生成器)