【python】yield用法入门

来源:互联网 发布:知乎平均985 编辑:程序博客网 时间:2024/06/07 03:08
包含yield表达式的函数是特殊的函数,叫做生成器函数(generator function),被调用时将返回一个迭代器(iterator),调用时可以使用next或send(msg)。它的用法与return相似,区别在于它会记住上次迭代的状态,继续执行。 
send(msg)与next()的区别在于send可以传递参数给yield表达式,这时传递的参数会作为yield表达式的值,而yield的参数是返回给调用者的值。初始调用时必须先next()或send(None),否则会报错。 

举个例子: 
1.png 
首先生成一个迭代器f,f.next()会使生成器函数执行到yield,生成一个值然后挂起。 
2.png 
然后f.next()或f.send(msg)会在生成器函数内部返回值,执行到下一个yield,生成值后挂起 
3.png 
然后f.next()或f.send(msg)会在生成器函数内部返回值,意图执行到下一个yield,但是后面没有yield了,所以抛出异常。 

使用yield可以有效简化代码,并减少空间浪费。 
举个简单例子:列表中的每个元素+1 
传统写法: 
Python代码
  1. def addlist(alist):   
  2.     r = []   
  3.     for i in alist:   
  4.         r.append(i+1)   
  5.     return r  
复制代码
yield写法: 
Python代码
  1. def addlist(alist):     
  2.     for i in alist:     
  3.         yield i + 1   
复制代码
当然对于这种简单的问题: 

Python代码
  1. [i+1 for i in alist]  
复制代码

参考文章

上面的笔记离不开下面文章的帮助,尤其是第一篇:

  • (译)Python关键字yield的解释(stackoverflow) — 一起写Python文章,一起看Python文章
  • Python3 yield使用 - Mr_Tank_ - 博客频道 - CSDN.NET
  • python 中 yield 用法的总结 - 简书
  • Python天天美味(25) - 深入理解yield - CoderZh - 博客