python的yield使用

在python中，yield被称之为generator（生成器）。

下面举个小例子来理解，yield在程序是怎么执行的

代码：

def func():
    print "begin"
    yield 1
    print "mid"
    yield 2
    print "end"
 
f = func()
f.next()  //1
f.next()  //2
f.next()  //3

f获得的是一个生成器（generator），next()每次都获取生成器中的下一个值。1的next会从函数的起始部分执行到第一个yield，也就是执行print “begin”和yield 1。所以语句1执行完输出begin，2执行完输出mid，3执行完输出end，并且会报错，报错是因为后面没有yield的了，当函数执行结束时，generator自动抛出StopIteration异常，表示迭代完成。在for循环里，无需处理StopIteration异常，循环会正常结束。

--------斐波那契数列------

知道了生成器的一些概念后，下面来看另一个例子

对于斐波那契数列，我们很容易想到下面的实现形式：

# coding=utf-8
 
def fab(n):
    index, a, b = 0, 0, 1
    res = []
    while index < n:
        res.append(b)
        a, b = b, a + b
        index = index + 1
    return res
 
for n infab(5):
    print n

使用yield的话，可以改写成如下形式：

# coding=utf-8
 
def fab(n):
    index, a, b = 0, 0, 1
    while index < n:
        # res.append(b)
        yieldb
        a, b = b, a + b
        index = index + 1
 
for n infab(5):
    print n

打印结果同前面那个程序的，如下：

原理就是第一次执行是从开始到yield b，yield b会返回b这个值。然后从后面执行到第二个yield b，以此类推。

 

------判断是否是generatorfunction------

使用isgeneratorfunction()函数判断一个函数是否是生成器函数

>>>from inspect import isgeneratorfunction
>>>isgeneratorfunction(fab)
True
 
>>>import types
>>>isinstance(fab, types.GeneratorType)
False
>>>isinstance(fab(5), types.GeneratorType)
True

区别fab和fab(5)，fab是一个generatorfunction，而fab(5)是调用fab返回的一个generator