程序员文章、书籍推荐和程序员创业信息与资源分享平台

网站首页 > 技术文章 正文

学会 Python yield 关键词,此文就够了

hfteth 2024-12-26 13:45:17 技术文章 24 ℃

???一、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
执行完成


???三、执行结果截图



???四、执行过程分析


  • 调用 test_generator 函数的时候,因为 test_demo 函数中有 yield 关键字,所以函数不会真的执行,而是先得到一个生成器 test_generator
  • 直到调用 next(test_generator) 方法,test_demo 函数开始正式执行,开始跑 for 循环
  • 当函数执行到第 15 行(result = yield num),此时 yield num 并没有赋值给 result,因此 16 行打印 result 的值永远为 None
  • 继续执行 19 行到 21 行顺序执行,通过 next(test_generator)将生成器的值输出
  • 如果没有继续调用 next(test_generator)整个函数执行完毕了
  • 顺序执行 22 行代码

???五、注意点

可以通过 send 方法给 result 重新赋值(result.send(100))

yield 后面也可以返回多个参数,多个参数返回类型为元组(通常在 pytest 前置后置中使用)



最近发表
标签列表