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)
输出如下: