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)语句执行完成。
4.程序执行print(""20),输出20个*
5.又开始执行下面的print(next(g)), 这个时候是从刚才那个next程序停止的地方开始执行的,也就是要执行res的赋值操作,要注意这个时候赋值操作的右边是没有值的(因为刚才那个是return出去了,并没有给赋值操作的左边传参数),所以这个时候res赋值是None,所以接着下面的输出就是res:None,
6.程序会继续在while里执行,又一次碰到yield,这个时候同样return 出4,然后程序停止,print函数输出的4就是这次return出的4.
yield的语法规则是:
在yield这里暂停函数执行,并返回yield后面表达式的值(默认为None),直到被next()再次调用时,从上次暂停的yield代码处继续往下执行。当没有可继续next()时,抛出异常,该异常可被for循环处理。
def fib(n):
a, b = 0, 1
i = 0
while i < n:
yield b
a, b = b, a+b
i += 1
if __name__ == '__main__':
f = fib(10)
for item in f:
print(item)
python中,含有yield关键字的对象就是一个生成器,每次调用next方法时会执行到yield后面的语句,然后返回yield后面代码块的执行结果
def foo():
bar_a = yield 1 # bar_a是语句块(yield 1)的返回值,默认为None
bar_b = yield bar_a
yield "最后一个值,再迭代就要报StopIteration了"
f = foo() # 创建生成器,此时没有执行foo()里的任何语句
print(next(f)) # 从foo()里进入,一直执行到(yield 1)处,此时变量bar_a还没有创建
print(next(f)) # 先将语句块(yield 1)的返回值赋值个bar_a,此时bar_a的值是None。
# 然后执行到语句块(yield bar_a),bar_b也还没有被创建
print(next(f)
>>>1
>>>None
>>>最后一个值,再迭代就要报StopIteration了
send()
def foo():
bar_a = yield 1
bar_b = yield bar_a
yield "最后一个值,再迭代就要报StopIteration了"
f = foo()
print(f.send(None)) # 和next(f)的效果一模一样
print(f.send("my lover")) # 覆盖yield 1语句的返回值
print(next(f))
>>>1
>>>my lover
>>>最后一个值,再迭代就要报StopIteration了