Python yield的用法详解

如何快速的对yield有一个初步的了解,那么首先我们可以先将yield看做是return,这样的话可以让我们有一个快速的对齐理解,然后我们还需要知道什么是生成器,因为当一个函数中出现了yield的关键字的时候返回的则是生成器

In [1]: def foo(x): 
   ...:     for i in range(x): 
   ...:         yield i  
   ...:         print(i) 
   ...:   
In [3]: foo(5)                                                                                             
Out[3]: <generator object foo at 0x7fcb054e9fc0>
In [4]: f = foo(5)                                                                                         
In [5]: next(f)                                                                                            
Out[5]: 0
In [6]: next(f)                                                                                            
0
Out[6]: 1
In [7]: next(f)                                                                                            
1
Out[7]: 2

代码的解释说明:
当程序开始执行的时候,因为foo函数中含有yield关键字,所以foo函数并不会真的执行,而是得到的一个生成器,对于生成器来说我们需要使用next的方法进行调用,所以当函数执行到yield的时候可以理解为return但是需要区分的是此时函数并没有执行完成,只不过是函数此时已经被挂起,等待下一步的调用.

关于生成器的send的用法

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

为什么会输出结果None,难道不是应该输出结果为4吗?其实关于这个问题就需要我们特别的明白和清楚就是yield从哪里进行将函数挂起,也就是说res = yield 4实际上从yield 4函数就不在执行了,也就是说并没有完成赋值操作.这个时候是从刚才那个next程序停止的地方开始执行的，也就是要执行res的赋值操作，这时候要注意，这个时候赋值操作的右边是没有值的（因为刚才那个是return出去了，并没有给赋值操作的左边传参数），所以这个时候res赋值是None,所以接着下面的输出就是res:None,
总结 :到这里你可能就明白yield和return的关系和区别了，带yield的函数是一个生成器，而不是一个函数了，这个生成器有一个函数就是next函数，next就相当于“下一步”生成哪个数，这一次的next开始的地方是接着上一次的next停止的地方执行的，所以调用next的时候，生成器并不会从foo函数的开始执行，只是接着上一步停止的地方开始，然后遇到yield后，return出要生成的数，此步就结束

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

starting...
4
********************
res: 7
4

先大致说一下send函数的概念：此时你应该注意到上面那个的紫色的字，还有上面那个res的值为什么是None，这个变成了7，到底为什么，这是因为，send是发送一个参数给res的，因为上面讲到，return的时候，并没有把4赋值给res，下次执行的时候只好继续执行赋值操作，只好赋值为None了，而如果用send的话，开始执行的时候，先接着上一次（return 4之后）执行，先把7赋值给了res,然后执行next的作用，遇见下一回的yield，return出结果后结束。
程序执行g.send(7)，程序会从yield关键字那一行继续向下运行，send会把7这个值赋值给res变量,由于send方法中包含next()方法，所以程序会继续向下运行执行print方法，然后再次进入while循环,程序执行再次遇到yield关键字，yield会返回后面的值后，程序再次暂停，直到再次调用next方法或send方法。

def bar():
    x = yield 1
    print(x)
    y = yield 2
    print(y)
b = bar()
print(next(b))  # x = yield 1 相当于 x = return 1 所以 返回结果1
print(b.send("X")) # 使用send发送字符串""X" 此时 x = "X" 函数继续向下执行 打印 X 
# 并且 遇到了 y = yeild 2 又相当于 y = return 2  所以函数打印 2 输出结果如下所示

1
X
2