前言

本文將由淺入深詳細介紹yield以及generator，包括以下內容：什么generator，生成generator的方法，generator的特點，generator基礎及高級應用場景，generator使用中的注意事項。本文不包括enhanced generator即pep342相關內容，這部分內容在之后介紹。

generator基礎

在python的函數（function）定義中，只要出現了yield表達式（Yield expression），那么事實上定義的是一個generator function， 調用這個generator function返回值是一個generator。這根普通的函數調用有所區別，For example：

def gen_generator(): yield 1def gen_value(): return 1 if __name__ == '__main__': ret = gen_generator() print ret, type(ret) #<generator object gen_generator at 0x02645648> <type 'generator'> ret = gen_value() print ret, type(ret) # 1 <type 'int'>

從上面的代碼可以看出，gen_generator函數返回的是一個generator實例

generator有以下特別：

     •遵循迭代器（iterator）協議，迭代器協議需要實現__iter__ 、next接口

     •能過多次進入、多次返回，能夠暫停函數體中代碼的執行

下面看一下測試代碼： 

>>> def gen_example():... print 'before any yield'... yield 'first yield'... print 'between yields'... yield 'second yield'... print 'no yield anymore'... >>> gen = gen_example()>>> gen.next()　　　　＃ 第一次調用nextbefore any yield'first yield'>>> gen.next()　　　　＃ 第二次調用nextbetween yields'second yield'>>> gen.next()　　　　＃ 第三次調用nextno yield anymoreTraceback (most recent call last): File "<stdin>", line 1, in <module>StopIteratio

調用gen example方法并沒有輸出任何內容，說明函數體的代碼尚未開始執行。當調用generator的next方法，generator會執行到yield 表達式處，返回yield表達式的內容，然后暫停（掛起）在這個地方，所以第一次調用next打印第一句并返回“first yield”。 暫停意味著方法的局部變量，指針信息，運行環境都保存起來，直到下一次調用next方法恢復。第二次調用next之后就暫停在最后一個yield，再次調用next()方法，則會拋出StopIteration異常。　

因為for語句能自動捕獲StopIteration異常，所以generator（本質上是任何iterator）較為常用的方法是在循環中使用：　

def generator_example(): yield 1 yield 2if __name__ == '__main__': for e in generator_example(): print e # output 1 2

generator function產生的generator與普通的function有什么區別呢

　　（1）function每次都是從第一行開始運行，而generator從上一次yield開始的地方運行