python笔记-3.1章_生成器

#自学Python的学习笔记

1.Generator

列表元素通过某种算法推算出来，在循环的过程中不断推算出后续的元素。这种一边循环一边计算的机制，称之为生成器

1.1创建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 0x02F0C1B0>

generator的值不能直接打印，需要通过next()函数获得，或者通过for循环打印

>>> next(g)
0
>>> next(g)
1
>>> next(g)
4
>>> next(g)
9
>>> next(g)
16
>>> next(g)
25
>>> next(g)
36
>>> next(g)
49
>>> next(g)
64
>>> next(g)
81
>>> next(g)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

每调用一次，返回一个generator的值；当所有值返回完毕后，再调用就会报错StopIteration

使用for循环

g = (x*x for x in range(10))
>>> for n in g:
	print(n)

	
0
1
4
9
16
25
36
49
64
81

 

斐波拉契数列（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
	return 'done'

>>> fib(8)
1
1
2
3
5
8
13
21
'done'

方法二

将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
	return 'done'

>>> fib(8)
<generator object fib at 0x02F2A570>

函数定义中包含yield关键字，这个函数就是一个generator

 

generator和函数的执行顺序不一样！

函数是顺序执行，遇到return或者最后一行函数语句就返回；

变成generator的函数，每次调用next()的时候执行，遇到yield语句返回，再次执行时从上次返回的yield语句处执行。