python的yield使用

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

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

代码：

def func():    print "begin"    yield 1    print "mid"    yield 2    print "end" f = func()f.next()  //1f.next()  //2f.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