Python：generator的send()方法流程分析

先来一个简单地例子：

def foo():
    print('starting')
    while True:
        r = yield 2
        print(r)

f = foo()
print(f.send(None))
print(f.send(1))

结果如下：

starting
2
1
2

 

讲解：

1   f = foo()  这句表示生成一个generator对象

f = foo()
print(type(f))  #<class 'generator'>

2   f.send(None) 的作用与 next(f) 的作用相同：运行代码到  r = yield 2 处。 r = yield 2 主要分两步：

　　第一步： yield 2 ，也就是先返回2

　　第二步： r = (yield) 这里用括号把yield包起来是为了突出yield是一个表达式expression：可以用来表示某个值。

 f.send(None) 或者说 next(f) 仅仅运行到了第一步，也就是返回了2，然后被print()函数打印到屏幕

3   f.send(1) 运行第二步，将1赋值给r ，然后运行print(r)，再一次运行到 r = yield 2 处时，也仅仅只运行第一步，也就是返回2，然后由print()函数打印到屏幕。

 

注意以下情况不会报错：

def foo():
    yield 1
    yield 2
    yield 3

f = foo()
r = next(f)
print(r)
for i in range(2):
    r = f.send('hello world')
    print(r)

# 结果
1
2
3

也就是说send非空值时，即使yield语句前没有变量接收send的值，程序不会报错