yield 关键字使用详解

发布于 更新于
本文目录
  1. 一、yield的作用
  2. 二、举个栗子,看下面这段代码
  3. 三、执行结果截图
  4. 四、执行过程分析
  5. 五、注意点

技术文章测试开发

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

一、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前置后置中使用)

来源:testingpai

继续阅读