什么是yield?

一句话理解:让普通函数,变成一个生成器(generator)[一个特殊的函数],函数里碰到yield就返回一次值。

yield 关键字就可以理解成和return一样功能,返回一个值。

生成器,生成器,就是使用了next才会生成一段yield的值。就是next指挥它跑一段,碰到yield就停下来,下次再从上次停的地方开始跑。 

 

一个例子:

def foo():
    print("starting...")
    while True:
        res = yield 4
        print("res:",res)
g = foo()
print(next(g))
print("*"*20)
print(next(g))
 

代码输出:

starting...             
4                       
********************    
res: None               
4                       

我直接解释代码运行顺序,相当于代码单步调试:

1.程序开始执行以后,因为foo函数中有yield关键字,所以foo函数并不会真的执行,而是先得到一个生成器g(相当于一个对象)

2.直到调用next方法,foo函数正式开始执行,先执行foo函数中的print方法,然后进入while循环

3.程序遇到yield关键字,然后把yield想想成return,return了一个4之后,程序停止,并没有执行赋值给res操作,此时next(g)语句执行完成,所以输出的前两行(第一个是while上面的print的结果,第二个是return出的结果)是执行print(next(g))的结果,

4.程序执行print("*"*20),输出20个*

5.又开始执行下面的print(next(g)),这个时候和上面那个差不多,不过不同的是,这个时候是从刚才那个next程序停止的地方开始执行的,也就是要执行res的赋值操作,这时候要注意,这个时候赋值操作的右边是没有值的(因为刚才那个是return出去了,并没有给赋值操作的左边传参数),所以这个时候res赋值是None,所以接着下面的输出就是res:None,

6.程序会继续在while里执行,又一次碰到yield,这个时候同样return 出4,然后程序停止,print函数输出的4就是这次return出的4.

 

总结:也就是有yield的函数,是生成器,生成器在调用的时候并不会真正执行【g = foo() 这样是不会真正执行】

只有调用了next(g),才会真正执行到碰到有yield的地方,然后就返回yield的值(比如yield 4)。

下一次再调用next(g)时,从刚刚yield返回的后面开始执行。直到下一次碰到next()

 

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

另一个例子:斐波那契(Fibonacci)數列是一个非常简单的递归数列,除第一个和第二个数外,任意一个数都可由前两个数相加得到。

计算斐波那契(Fibonacci)數列。

#!/usr/bin/python
# -*- coding: UTF-8 -*-
 
def fab(max): 
    n, a, b = 0, 0, 1 
    while n < max: 
        yield b      # 使用 yield
        # print b 
        a, b = b, a + b 
        n = n + 1
 
for n in fab(5): 
    print n

在 for 循环中会自动调用 next()。

当函数执行结束时,generator 自动抛出 StopIteration 异常,表示迭代完成。在 for 循环里,无需处理 StopIteration 异常,循环会正常结束。

也可以手动调用next()

>>>f = fab(5) 
>>> f.next() 
1 
>>> f.next() 
1 
>>> f.next() 
2 
>>> f.next() 
3 
>>> f.next() 
5 
>>> f.next() 
Traceback (most recent call last): 
 File "<stdin>", line 1, in <module> 
StopIteration

 

posted @ 2020-12-08 16:18  紫电清霜  阅读(940)  评论(0编辑  收藏  举报