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.程序占用内存少