Python3中 yield 关键字用法

今天在看Python书籍的时候，有段代码中用到了 yield 关键字，不知道作什么用，不甚理解。查了些资料，觉得 yield 挺好用的，给大家分享下：

在Python3中任何使用到 yield 的函数称为生成器，如：

def count(n):
    while n > 0:
        yield n  # 生成值： n
        n -= 1

另外一种说法：生成器就是一个返回迭代器的函数，与普通函数的区别是生成器包含yield语句，更简单点理解生成器就是一个迭代器。

使用yield，可以让函数生成一个序列，该函数返回的对象类型是"generator"，通过该对象连续调用__next__()方法返回序列值。

def count(n):
    while n > 0:
        yield n  # 生成值： n
        n -= 1


c = count(5) 
print(c.__next__())  # 输出：5
print(c.__next__())  # 输出：4

生成器函数只有在调用__next()__方法的时候才开始执行函数里面的语句，比如：增加打印输出

def count(n):
    print("bgn...")
    while n > 0:
        yield n  # 生成值： n
        n -= 1

在调用count函数时：c=count(5)，并不会打印"bgn..."。只有等到调用c.__next__()时才真正执行里面的语句。每次调用__next__()方法时，count函数会运行到语句yield n处为止，__next__()的返回值就是生成值n，再次调用__next__()方法时，函数继续执行yield之后的语句。

从上面输出可以看出，调用了2次 c.__next__，“bgn...”只输出一次，“end...”没有输出。“end...”只有在迭代器的值都迭代完成才会输出。如果一直调用__next__方法，当执行到没有可迭代的值后，程序就会报错：

Traceback (most recent call last):

所以一般不会手动的调用__next__方法，而使用for循环，完整代码如下：

def count(n):
    print("bgn...")
    while n > 0:
        yield n  # 生成值： n
        n -= 1
    print("end...")


for i in count(5):
    print(i)

输出如下：