包含了關鍵字"yield"的函數就不是普通的函數。當含有這個關鍵字的函數被調用的時候,這個函數在遇到yield的時候會停止運行,並且返回一個迭代器(iterator)。每次請求一個值,就會執行產生的程式碼。直到遇到一個yield或者return。
首先,我們先瞭解什麼是迭代器。
lst=[1,2,3,4,5]
for i in lst
print i
從這個例子我們可以看的出來,每迴圈一次i的值就會指向列表的下一個元素,大家認為這是正常的,那麼為什麼i會得到列表的下一個元素呢?
其實在for的迴圈中列表就使用了迭代器。每一次迴圈迭代器就使用next方法返回一個值。當然這個迭代是隱形的,大家是看不見的。
我們可以實現一個可迭代的函數。
#!/ust/bin/env python
class IterExample():
def __init__(self):
self.a = 0
def next(self):
self.a += 1
if self.a > 10:raise StopIteration
return self.a
def __iter__(self):
return self
ie = IterExample()
for i in ie:
print i
上面的列表預設已經具備了迭代方法,不用我們實現。如果一個函數不是可迭代的那是不能用在迴圈裡的。
下面我們來解決yield
其實這個很簡單。只不過大家看的例子複雜了。看看下面的例子你立刻就會理解它。
#!/usr/bin/env python
#__metaclass__ = type
def gen():
print 'enter'
yield 1
print 'next'
yield 2
print 'next again'
for i in gen():
print i
#########################
這個例子列印如下:
enter
1
next
2
next again
我來解釋一下這個程式:
首先大家要知道為什麼這個函數可以用在for迴圈中。不用問,因為這個函數是可迭代的,也就是這個函數可以每次都返回一個值。
但是我們在gen()函數裡看不到__iter__()和next() 方法。其實它隱藏在yield裡。進階語言就是這樣,隱藏了好多東西。這個和c
語言就不同了。C語言可以看到每個細節。在這裡程式執行到yield 1的時候就停止了,下面的程式不再執行,然後返回一個值“1”.
當下一個for的之後程式接著往下執行到yeild 2。程式停止執行,然會返回一個值“2”。但是還有一個問題就是,最後yield下面的"next again" 會列印出來呢?這也可能就是在執行完最後一個yield 的時候,for i in gen()的時候, 發生了一些什麼動作,導致最後一個yield後面的代碼也被執行了。
如果我們修改一下這個程式如下:
#!/usr/bin/env python
#__metaclass__ = type
def gen():
print 'enter'
yield 1
print 'next'
return
print 'next 2'
yield 2
print 'next 3'
for i in gen():
print i
#######################
程式列印如下:
enter
1
next
看明白了這就是yield和return的區別。yield可以向下運行。而return返回後這個函數餘下的部分就不能執行了。