python yield原理总结

生成器

生成器是包含关键字 yield的函数，但被调用时不会执行函数体内的代码，而是返回一个迭代器。

生成器由两个单独的部分组成：生成器的函数和生成器的迭代器。生成器的函数 是由def语句定义的，其中包含yield。

迭代器

生成器的迭代器是生成器的函数返回的结果。

注意：

更正规的定义是，实现了方法__iter__的对象是可迭代的，而实现了方法__next__的对象 是迭代器。

在Python 3中，迭代器协议有细微的变化。在以前的迭代器协议中（python2），要求迭代器对象包含方法next而不是__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。

 

再看一个这个生成器的send函数的例子，

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出结果后结束。

 

5.程序执行g.send(7)，程序会从yield关键字那一行继续向下运行，send会把7这个值赋值给res变量

6.由于send方法中包含next()方法，所以程序会继续向下运行执行print方法，然后再次进入while循环

7.程序执行再次遇到yield关键字，yield会返回后面的值后，程序再次暂停，直到再次调用next方法或send方法。

注意：

生成器刚启动时（第一次调用），必须使用next()语句或是send(None)，不能直接发送一个非None的值，否则会报TypeError，可以认为next()等同于send(None)，(事实上，如上所说，send会调用next())，也就是说当程序没有停在yield语句之后（或者说没有从yield语句返回），那么就不能接收send所发送的参数，所以只能使用send(None)或者next()。

简单一句话，next()不会给yield语句之前的变量赋值，而send可以，send带的参数就是为了带给yield语句=之前的待赋值变量。send(x) = next()+把x赋值给目标。

 

参考链接：

《python基础教程（第3版）》

https://blog.csdn.net/mieleizhi0522/article/details/82142856/

https://www.cnblogs.com/ellisonzhang/p/10273843.html

https://www.jianshu.com/p/9dd355ab4e5d