当前位置:   article > 正文

python中yield关键字方法详解_python yield

python yield

        yield 就是保存当前程序执行状态。你用 for 循环的时候,每次取一个元素的时候就会计算一次。用 yield 的函数 叫 generator,和 iterator 一样,它的好处是不用一次计算所有元素,而是用一次算一次,可以节省很多空间。generator 每次计算需要上一次计算结果,所以用 yield,否则一 return,上次计算结果就没了。

        yield可以简单理解为return操作,但和return又有很大的区别,执行完return,当前函数就终止了,函数内部的所有数据,所占的内存空间,全部都没有了。而yield在返回数据的同时,还保存了当前的执行内容,当你再一次调用这个函数时,他会找到你在此函数中的yield关键字,然后从yield的下一句开始执行。

        当有多个返回值时,用 return 全部一起返回了,需要单个逐一返回时可以用 yield。

        带有yield的函数在Python中被称之为generator(生成器),也就是说,当你调用这个函数的时候,函数内部的代码并不立即执行 ,这个函数只是返回一个生成器(Generator Iterator)。

示例代码1:

  1. def generator():
  2. for i in range(10):
  3. yield i * i
  4. gen = generator()
  5. print(gen)

示例代码2:

  1. def generator():
  2. for i in range(10):
  3. yield i * i
  4. gen = generator()
  5. print(gen)
  6. print("first:")
  7. print(next(gen))
  8. print("second:")
  9. print(next(gen))

 注意:

  • 在函数第一次调用next(gen)函数时,generator函数从开始执行到yield,并返回yield之后的值。
  • 在函数第二次调用next(gen)函数时,generator函数从上一次yield结束的地方继续运行,直至下一次执行到yield的地方,并返回yield之后的值。依次类推。

yield 关键字:

  • 只要在def函数里面看到有 yield 关键字那么就是生成器

示例代码1:

  1. def mygenerater(n):
  2. for i in range(n):
  3. print('开始生成...')
  4. yield i
  5. print('完成一次...')
  6. if __name__ == '__main__':
  7. g = mygenerater(2)
  8. # 获取生成器中下一个值
  9. result = next(g)
  10. print(result)
  11. result = next(g)
  12. print(result)

 示例代码2:

  1. def mygenerater(n):
  2. for i in range(n):
  3. print('开始生成...')
  4. yield i
  5. print('完成一次...')
  6. if __name__ == '__main__':
  7. g = mygenerater(3)
  8. # 获取生成器中下一个值
  9. # result = next(g)
  10. # print(result)
  11. # result = next(g)
  12. # print(result)
  13. #
  14. while True:
  15. try:
  16. result = next(g)
  17. print(result)
  18. except StopIteration as e:
  19. break

 示例代码3:

  1. def mygenerater(n):
  2. for i in range(n):
  3. print('开始生成...')
  4. yield i
  5. print('完成一次...')
  6. if __name__ == '__main__':
  7. g = mygenerater(3)
  8. # 获取生成器中下一个值
  9. # result = next(g)
  10. # print(result)
  11. # result = next(g)
  12. # print(result)
  13. #
  14. # while True:
  15. # try:
  16. # result = next(g)
  17. # print(result)
  18. # except StopIteration as e:
  19. # break
  20. #
  21. # for遍历生成器, for 循环内部自动处理了停止迭代异常,使用起来更加方便
  22. for i in g:
  23. print(i)

代码说明:

  • 代码执行到 yield 会暂停,然后把结果返回出去,下次启动生成器会在暂停的位置继续往下执行
  • 生成器如果把数据生成完成,再次获取生成器中的下一个数据会抛出一个StopIteration 异常,表示停止迭代异常
  • while 循环内部没有处理异常操作,需要手动添加处理异常操作
  • for 循环内部自动处理了停止迭代异常,使用起来更加方便,推荐大家使用。

函数中含有多个yield关键字:

  • 一次只能返回一个,多个yield关键字返回的内容轮流一次返回

示例代码1:

  1. def mygenerater(n):
  2. for i in range(n):
  3. print('开始生成...')
  4. yield i
  5. print('完成一次...')
  6. yield i+100
  7. print('完成二次...')
  8. print("生成一轮结束")
  9. if __name__ == '__main__':
  10. g = mygenerater(3)
  11. # 获取生成器中下一个值
  12. result = next(g)
  13. print(result)
  14. result = next(g)
  15. print(result)

 示例代码2:

  1. def mygenerater(n):
  2. for i in range(n):
  3. print('开始生成...')
  4. yield i
  5. print('完成一次...')
  6. yield i+100
  7. print('完成二次...')
  8. print("生成一轮结束")
  9. if __name__ == '__main__':
  10. g = mygenerater(3)
  11. # 获取生成器中下一个值
  12. result = next(g)
  13. print(result)
  14. result = next(g)
  15. print(result)
  16. result = next(g)
  17. print(result)

示例代码3:

  1. def mygenerater(n):
  2. for i in range(n):
  3. print('开始生成...')
  4. yield i
  5. print('完成一次...')
  6. yield i+100
  7. print('完成二次...')
  8. print("生成一轮结束")
  9. if __name__ == '__main__':
  10. g = mygenerater(3)
  11. # 获取生成器中下一个值
  12. result = next(g)
  13. print(result)
  14. result = next(g)
  15. print(result)
  16. result = next(g)
  17. print(result)
  18. result = next(g)
  19. print(result)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/42253
推荐阅读
相关标签
  

闽ICP备14008679号