python中yield的用法

Posted 2021-01-31 努力奋斗的阿贝拉

---"在python中，当你定义一个函数，使用了yield关键字时，这个函数就是一个生成器" (也就是说，只要有yield这个词出现，你在用def定义函数的时候，系统默认这就不是一个函数啦，是一个生成器啦!!不管里面内容你写成神马样子 )

--- 一般def定义的函数，都会return一个返回值。而def定义的生成器，返回的则是一个对象，也就是上面提到的类似于“内存地址”的东西。（看来我上面解释的还不是很对）

--- 如果需要生成器返回（下一个）值，需要调用.next()函数。其实当系统判断def是生成器时，就会自动支持.next()函数

 

c = h() #h()包含了yield关键字    
print  c.next() #返回值     

 ----------------------------------------------------------------------------------------------------------------

辣么，我们来做一个小小的习题，看看是否理解了yield

----------------------------------------------------------------------------------------------------------------

 

def fib(max):
    a, b = 1, 1  
    while a < max:  
        yield a  
        a, b = b, a+b  
  
for n in fib(15):  
    print n  
  
m = fib(13)  
print m  
print m.next()  
print m.next()  
print m.next()  

 

以上是一段关于斐波那契数列的程序。

fib()函数因为含有yield，被系统默认为是一个生成器。

for语句调用了fib(15)。当max=15时，进入fib()生成器，执行到yield a， 返回a值以及整个生成器暂停的状态，将a值赋给n， 打印出来；因为是for语句循环，所以又回到fib(15)语句，由于是生成器，因此从上次截断的位置开始执行，b值赋给a, a+b值赋给b，又因为是while语句，则继续while循环，yield a值，循环暂停跳出返回a值及生成器状态，把a值赋给n, 打印n。如此往复，一直循环到15结束。

m被赋了fib(13)这个生成器，每一次执行m.next()函数就会打印下一个值。

从上面的分析过程，我们看一下运行结果：

 

注意：

1. 每个生成器只能使用一次。比如上个例子中的m生成器，一旦打印完m的6个值，就没有办法再打印m的值了，因为已经吐完了。

2. yield一般都在def生成器定义中搭配一些循环语句使用，比如for或者while，以防止运行到生成器末尾跳出生成器函数，就不能再yield了。有时候，为了保证生成器函数永远也不会执行到函数末尾，会用while True: 语句，这样就会保证只要使用next()，这个生成器就会生成一个值，是处理无穷序列的常见方法。