Python中yield的用法

Python中yield的用法

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

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

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

例1:

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

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

例2:

>>> 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:

>>> 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