Python中yield的用法

来源：互联网发布：js设为首页和加入收藏编辑：程序博客网时间：2024/05/21 12:08

yield是生成的意思，但是在python中则是作为生成器理解，生成器的用处主要可以迭代，这样简化了很多运算模型(还不是很了解是如何简化的)。

yield是一个表达式,是有返回值的.

当一个函数中含有yield时,它不再是一个普通的函数,而是一个生成器.当该函数被调用时不会自动执行,而是暂停,见第一个例子:

例1:

[python] view plain copy
>>> def mygenerator():  
...     print 'start...'  
...     yield 5  
...   
>>> mygenerator()            //在此处调用,并没有打印出start...说明存在yield的函数没有被运行,即暂停  
<generator object mygenerator at 0xb762502c>  
>>> mygenerator().next()     //调用next()即可让函数运行.  
start...  
5  
>>>   

如一个函数中出现多个yield则next()会停止在下一个yield前,见例2:

例2:

[python] view plain copy
>>> def fun2():  
...     print 'first'  
...     yield 5  
...     print 'second'  
...     yield 23  
...     print 'end...'  
...   
>>> g1 = fun2()  
>>> g1.next()             //第一次运行,暂停在yield 5               
first  
5  
>>> g1.next()             //第二次运行,暂停在yield 23  
second  
23  
>>> g1.next()             //第三次运行,由于之后没有yield,再次next()就会抛出错误  
end...  
Traceback (most recent call last):  
  File "<stdin>", line 1, in <module>  
StopIteration  
>>>   

为什么yield 5会输出5,yield 23会输出23?

我们猜测可能是因为yield是表达式,存在返回值.

那么这是否可以认为yield 5的返回值一定是5吗？实际上并不是这样，这个与send函数存在一定的关系，这个函数实质上与next()是相似的，区别是send是传递yield表达式的值进去，而next不能传递特定的值，只能传递None进去，因此可以认为g.next()和g.send(None)是相同的。见例3:

例3:

[python] view plain copy
>>> def fun():  
...     print 'start...'  
...     m = yield 5  
...     print m  
...     print 'middle...'  
...     d = yield 12  
...     print d  
...     print 'end...'  
...   
>>> m = fun()              //创建一个对象  
>>> m.next()               //会使函数执行到下一个yield前  
start...  
5  
>>> m.send('message')      //利用send()传递值  
message                    //send()传递进来的   
middle...  
12  
>>> m.next()  
None                       //可见next()返回值为空  
end...  
Traceback (most recent call last):  
  File "<stdin>", line 1, in <module>  
StopIteration  

0 0