【python】yield详细用法

首先，如果你还没有对yield有个初步分认识，那么你先把yield看做“return”，这个是直观的，它首先是个return，普通的return是什么意思，就是在程序中返回某个值，返回之后程序就不再往下运行了。而yield 与return有些区别，那先看下面的程序，先初步yield的使用：

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

到这里你可能就明白yield和return的关系和区别了，带yield的函数是一个生成器（generator保存的是算法），而不是一个函数了，这个生成器有一个函数就是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))
#>>>4
print("*"*20)
print(next(g))
#>>>res: None
#>>>4

程序分析：

程序开始执行以后，因为foo函数中有yield关键字，那么这个函数编程生成器, 对其进行函数调用, ，不会执行, 会返回一个生成器对象。所以foo函数并不会真的执行，而是先得到一个生成器g(相当于一个对象)
直到调用next方法，foo函数正式开始执行，先执行foo函数中的print方法，然后进入while循环
程序遇到yield关键字，然后把yield想成return,return了一个4之后，程序停止，并没有执行赋值给res操作，此时next(g)语句执行完成，所以输出的前两行（第一个是while上面的print的结果,第二个是return出的结果）是执行print(next(g))的结果。
程序执行print("*"*20)，输出20个*
又开始执行下面的print(next(g)),这个时候和上面那个差不多，不过不同的是，这个时候是从刚才那个next程序停止的地方开始执行的，也就是要执行res的赋值操作，这时候要注意，这个时候赋值操作的右边是没有值的（因为刚才那个是return出去了，并没有给赋值操作的左边传参数），所以这个时候res赋值是None,所以接着下面的输出就是res:None。
程序会继续在while里执行，又一次碰到yield,这个时候同样return 出4，然后程序停止，print函数输出的4就是这次return出的4.

如何才能实现给res赋值呢，再看一个这个生成器的send函数的例子：

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

send函数的概念：此时你应该注意到上面那个res的值为什么是None，这个变成了7，到底为什么？这是因为，send是发送一个参数给res的，因为上面讲到，return的时候，并没有把4赋值给res，下次执行的时候只好继续执行赋值操作，只好赋值为None了，而如果用send的话，开始执行的时候，先接着上一次（return 4之后）执行，先把7赋值给了res,然后执行next的作用，遇见下一回的yield，return出结果后结束。

发现, send()和next()都可以让生成器向下执行, 不同的是, send(arg)通过一个参数, 给yield res整体赋值.

需要注意的是, 在生成器的第一次使用时, 由于程序并未执行到yield, 所以此时通过send()函数传参调用生成器会报错TypeError: can't send non-None value to a just-started generator, 但是可以传入参数None 来使生成器走到yield处