深入理解Python生成器（Generator）(2)

当前位置:

首页 > Python基础教程 >

深入理解Python生成器（Generator）(2)

上面的函数可以输出斐波那契数列的前N个数：

1

2

3

4

5

6

7

仔细观察，可以看出，fib函数实际上是定义了斐波拉契数列的推算规则，可以从第一个元素开始，推算出后续任意的元素，这种逻辑其实非常类似generator。

也就是说，上面的函数和generator仅一步之遥。要把fib函数变成generator，只需要把print b改为yield b就可以了：

1

2

3

4

5

6

7

						def fib(max):

						    n = 0

						    a, b = 0, 1

						    while n < max:

						        yield b

						        a, b = b, a + b

						        n = n + 1

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

1 2	`>>> fib(6)` `<generator object fib at 0x104feaaa0>`

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

举个简单的例子，定义一个generator，依次返回数字1，3，5：

						>>> def odd():

						...     print 'step 1'

						...     yield 1

						...     print 'step 2'

						...     yield 3

						...     print 'step 3'

						...     yield 5

						...

						>>> o = odd()

						>>> o.next()

						step 1

						1

						>>> o.next()

						step 2

						3

						>>> o.next()

						step 3

						5

						>>> o.next()

						Traceback (most recent call last):

						  File "<stdin>", line 1, in <module>

						StopIteration

栏目列表