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

时间:2024-06-18 23:34:56

先来一个简单地例子:

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的值,程序不会报错