yield 关键字使用详解

本贴最后更新于 1100 天前,其中的信息可能已经渤澥桑田
一、yield的作用

yield 在函数中相当于return,又不同于return,当函数执行遇到yield的时候,函数会停止执行,并返回yield 后值

二、举个栗子,看下面这段代码
def test_demo():
    for num in range(1,4):
        print('starting..........')
        result = yield num
        print('yield num并没有赋值给result,result的值为None,result = ', result)

test_generator = test_demo()
print("这里是生成器:", test_generator)
print("生成器中第1个值:", next(test_generator))
print("生成器中第2个值:", next(test_generator))
print("生成器中第3个值:", next(test_generator))
print("执行完成")

执行结果
这里是生成器 <generator object test_demo at 0x000001728B817748>
starting..........
生成器中第1个值 1
yield num并没有赋值给result,result的值为None,result =  None
starting..........
生成器中第2个值 2
yield num并没有赋值给result,result的值为None,result =  None
starting..........
生成器中第3个值 3
执行完成
三、执行结果截图

image.png

四、执行过程分析
  1. 调用test_generator函数的时候,因为test_demo函数中有yield关键字,所以函数不会真的执行,而是先得到一个生成器test_generator
  2. 直到调用next(test_generator) 方法,test_demo函数开始正式执行,开始跑for循环
  3. 当函数执行到第15行(result = yield num),此时yield num并没有赋值给result,因此16行打印result的值永远为None
  4. 继续执行19行到21行顺序执行,通过next(test_generator)将生成器的值输出
  5. 如果没有继续调用next(test_generator)整个函数执行完毕了
  6. 顺序执行22行代码
五、注意点
  1. 可以通过send方法给result重新赋值(result.send(100))
  2. yield 后面也可以返回多个参数,多个参数返回类型为元组(通常在pytest前置后置中使用)
回帖
请输入回帖内容 ...