Python3生成器的执行流程详解

Python3 生成器的执行流程详解

生成器，英文名称generator,形式有两种：

1.自定义函数，def的函数区块里加上yield，遇到yield则停止。

2.列表生成器：将列表生成式的[ ]，改成( )，即可。

生成器的执行有点难以理解，下面进行解释说明：

首先，上栗子：

1   def consumer():
2       r='here'     
3       while True:
4           n=yield r
5           if not n:
6               return
7           print('[Consumer]Consuming %s...'%n)
8           r='200 OK' +str(n)
9
10  def produce(c):
11      c.send(None)
12      n=0
13      while n<5:
14        n=n+1
15        print('[Producer]Producing %s...' %n)
16        r=c.send(n)
17        print('[Producer]Consumer return: %s' %r)
18      c.close()
19
20  c=consumer()
21  produce(c)

运行结果如下：

[Producer]Producing 1...
[Consumer]Consuming 1...
[Producer]Consumer return: 200 OK1
[Producer]Producing 2...
[Consumer]Consuming 2...
[Producer]Consumer return: 200 OK2
[Producer]Producing 3...
[Consumer]Consuming 3...
[Producer]Consumer return: 200 OK3
[Producer]Producing 4...
[Consumer]Consuming 4...
[Producer]Consumer return: 200 OK4
[Producer]Producing 5...
[Consumer]Consuming 5...
[Producer]Consumer return: 200 OK5

Process finished with exit code 0

运行原理如下：

1. 首先是执行第20行，第21行。从第21行传入参数到第10行，继续执行第11行，遇到c.send(None)，.send是生成器的启动方法，表示启动的生成器。
2. 程序跳转，开始执行第1行，一直到第四行，到yield前面为止，第四行不执行，也就是不赋值给n。
3. 程序继续执行第12行，0赋值给n。
4. 第13行，进入while循环。第14行，n+1复制给n。第15行，打印print ('[Producer]Producing %s...' %n)。
5. 执行第16行，遇到c.send(n)，启动生成器，跳转到程序第4行，将n赋值给n。继续执行第5行，到8行，由于while循环，继续执行第4行，遇到yield，停止执行，返回r值，是'200 OK' +str(n)。
6. 执行第16行，将r赋值给r。继续执行第17行，由于while循环，从第13行执行一直到第16行，再次遇到c.send(n)，启动生成器，跳转程序至第4行，将n赋值给n。
7. 重复执行第4步和第5步。一直到n值等于5，跳出while循环，执行第15行。遇到c.close()，关闭生成器。函数执行结束。
8. 程序执行结束。

参考：

廖雪峰的python3教程

Python生成器generator之next和send运行流程