send python_python 生成器 send()方法简介

#!/usr/bin/python3

def MyGenerator():

value=yield 1

yield value

return done

gen=MyGenerator()

print(next(gen))

print(gen.send("I am Value"))

生成器内有一个方法send，可再次传入一个值。

上面那句可能听不懂，但是不要紧，我们先来看看代码，

#!/usr/bin/python3

def MyGenerator():

value=yield 1

yield value

return done

gen=MyGenerator()

print(next(gen))

print(gen.send("I am Value"))

代码分析，

在MyGenerator里，我们一共用了两次yield。

比较奇怪的是第一个yield的语句，value=yield 1。如果没看过这一语句的，肯定不知道next回到yield后，其实是有一个值的。

到这，我们先不急，运行结果，

[penx@ali01 python]$ ./gen_send.py

1

I am Value

[penx@ali01 python]$

运行过程，

用next启动了生成器gen，知道到yield 1时返回1。

然后我们再用gen的内部方法send进入gen，而且还带回来一个值“I am Value”。这时候，继续执行yield 1后的代码“value=”，把带回来的值“I am Value”赋给value。直到遇到yield value，把value返回。

其实，send和next的执行很像，只是send可以和生成器互动，传入一个值。

生成器的启动需要next

大家有没有想过，如果生成器还没启动过，就用send，会怎样？我们来试一下。

代码，

#!/usr/bin/python3

def MyGenerator():

value=yield 1

yield value

return done

gen=MyGenerator()

print(gen.send(3))

运行，

[penx@ali01 python]$ ./gen_send.py

Traceback (most recent call last):

File "./test.py", line 9, in

print(gen.send(3))

TypeError: can't send non-None value to a just-started generator

[penx@ali01 python]$

结果，

报错，

> TypeError: can’t send non-None value to a just-started generator

说生成器刚启动时，不能send一个不为None的值。

小结，

所以呢，我们在用生成器时，第一次要用next启动

生成器启动可用send(None)

其实上面报错已经说了，can’t send non-None value

所以啊，我们可以用send(None)来启动生成器。

代码，

#!/usr/bin/python3

def MyGenerator():

value=yield 1

yield value

return done

gen=MyGenerator()

print(gen.send(None))

print(gen.send(3))`

运行，

[penx@ali01 python]$ ./gen_send.py

1

3

[penx@ali01 python]$

结果，

正常运行。