python yield生成器函数

标签：code   包含   max   coding   print   while   next   包括   方法   

yield是python的一个关键字，本质上是一个生成器generator。

生成器是一种特殊的函数，它会返回一个迭代器。定义一个生成器函数同定义一个普通函数没有什么区别，特殊之处在于生成器函数内部会包含yield表达式，专门用于生成一个序列。当一个生成器函数被调用时，它会返回一个迭代器。之后就由这个迭代器来控制生成器函数的执行。当生成器函数被调用后，首先会执行到第一个yield表达式处，然后会将生成器函数挂起，将yield生成的表达式的值返回给生成器函数的调用者。当生成器函数被挂起时，它的所有局部状态都会被保存起来，包括当前绑定的局部变量、指令指针、内部栈和异常处理的状态。当调用迭代器的方法时，一般都是调用next()方法，将会恢复生成器函数的执行，并且是从上次被挂起的地方继续执行，直到遇到另外一次yield调用，生成器函数将再次被挂起。

在一个生成器函数中，如果没有 return，则默认执行至函数完毕，如果在执行过程中 return，则直接抛出 StopIteration 终止迭代。

下面通过斐波那契数列的经典案例来体系yield的优秀特性。实际上函数fab已经不是一个普通函数，而是一个生成器。

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

def fab(max):
    n, a, b = 0, 0, 1
    while n < max:
        yield b      # use yield，立即返回，下次从后面继续执行，这就是yield强大的地方
        a, b = b, a + b
        n = n + 1

for n in fab(5):
    print n

print ‘------------------‘

a = fab(5)
print a.next()
print a.next()
print a.next()
print a.next()
print a.next()
print a.next() #exception, but for handle it，此处抛出异常StopIteration，但是for循环处理了这个异常。

使用yield的优点：

1.代码整洁
2.程序占用内存少

python yield生成器函数

标签：code   包含   max   coding   print   while   next   包括   方法   

原文地址：https://blog.51cto.com/14207158/2524749