天天看点

[Python]yield用法入门

包含yield表达式的函数是特殊的函数,叫做生成器函数(generator function),被调用时将返回一个迭代器(iterator),调用时可以使用next或send(msg)。它的用法与return相似,区别在于它会记住上次迭代的状态,继续执行。 

send(msg)与next()的区别在于send可以传递参数给yield表达式,这时传递的参数会作为yield表达式的值,而yield的参数是返回给调用者的值。初始调用时必须先next()或send(None),否则会报错。 

举个例子: 

[Python]yield用法入门

首先生成一个迭代器f,f.next()会使生成器函数执行到yield,生成一个值然后挂起。 

[Python]yield用法入门

然后f.next()或f.send(msg)会在生成器函数内部返回值,执行到下一个yield,生成值后挂起 

[Python]yield用法入门

然后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]  

复制代码