Python中的yield,幾個例子和理解

yield 實現Fibonacci數列

實現Fibonacci數列是生成器的經典例子:

def fab(max):  
n, a, b = 0, 0, 1  
while n < max:  
        yield b  
        # print b  
        a, b = b, a + b  
        n = n + 1 

代碼是如此的簡單,但想要深入了解yield并非表面上這么簡單。其實yield是個語法糖,其作用就是把一個函數變成一個 generator類,因此帶有 yield 的函數不再是一個普通函數,Python 解釋器會將其視為一個 generator,調用 fab(5) 不會執行 fab 函數,而是返回一個 可迭代的對象在 for 循環執行時,每次循環都會執行 fab 函數內部的代碼,執行到 yield b 時,fab 函數就返回一個迭代值,下次迭代時,代碼從 yield b 的下一條語句繼續執行,而函數的本地變量看起來和上次中斷執行前是完全一樣的,于是函數繼續執行,直到再次遇到 yield。

進一步地深入:next()和send()

寫一個更簡單的例子來說明問題:

def h(): 
        print "hello" 
        yield 5 
        print "world"

如前面所說,在寫這個函數的時候,由于使用了yield,使得改函數變為一個生成器類,而生成器自帶.next()方法。將其實例化:
c = h()
則它帶有next()方法。在每次執行c.next()的時候,python會依次執行函數h()體內的語句,直到出現一個yield,程序返回yield 后的內容,并會掛起。直到下一次執行c.next()。重復上述過程,直到后面沒有yield語句時,會拋出StopInteration異常,如果迭代發生在for循環中,相應地,循環會結束。因此有以下的執行結果(# 后為輸出結果):

c.next()  # hello   # 5
c.next()  # world   # StopIteration

進一步,可以將 yield看作表達式,并賦值給一個變量,如下面的函數:

def h(): 
        print "hello" 
        m = yield 5 
        print "m:", m 
        d = yield 12 
        print "d:", d 
        print "world"

那么問題來了:這里的md的值是什么呢?下面來看運行結果:

c = h()
c.next()  # hello   # 5
c.next()  # m: None   # 12
c.next()  # d: None   # world   # StopIteration

這里第一次調用next()時候返回的5是第一個yield 5的結果。可以看出,第二次掉用next()時返回的None就是變量m的值。那m的值為什么是None?其實生成器還有一個內建的send()方法,且send()方法和next()方法的功能有很大的重合:都是讓函數體繼續向下執行直到遇到下一個yield語句并掛起。而不同點是send(msg)方法會給生成器發送變量msg,并作為yield表達式的返回值賦值給其前面的變量,也就是h()中的m和d.

因此,next()就等價于send(None)。我們再用send方法來運行上面的函數:

c = h()
c.send("I am m") 
# TypeError: can't send non-None value to a just-started generator

可以看到出錯了,正如錯誤提示所說,對于剛開始的生成器是不能send 一個non-None值的。由此也可以知道,必須要先yield,才能send(msg)。再來:

c = h()
c.send(None) # hello   # 5
c.send("I am m") # m: I am m   # 12
c.send("I am d")  # d: I am d   # world   # StopIteration

由上可以總結出,yield 后面的內容和前面的賦值其實沒什么關系,前面的賦值是在yield 完成以后由send()決定的。

有了上述知識,再看一個稍微復雜的例子加深理解:

def countdown(n): 
        print "Counting down from", n 
        while n >= 0: 
                newvalue = (yield n) 
                # If a new value got sent in, reset n with it 
                if newvalue is not None: 
                        n = newvalue 
                else: 
                        n -= 1

c = countdown(10)
for x in c: 
        print x 
        if x == 10: 
                c.send(3)

輸出結果是:

10
2
1
0

在for循環的第一次,相當于執行了c.next(),程序往下執行直到yield 10,因此x得到了值10且程序掛起,等待下一次的next()或send()調用。 此時,由于x=10, 在for循環中又會執行一次c.send(3),剛才掛起的程序將會繼續往下執行:因為send的是3,上一次被yield表達式賦值的變量newvalue將會被賦值為3,因此在接下來的if語句中,n被賦值為3. 接著進入下一輪while循環,yield出了n=3,且程序被掛起。注意:這一次yield n并不是由for循環引起的(而是c.send()引起的),因此x并沒有得到3這個值。接著才是下一輪for循環,相當于又掉用了c.next(),程序向下執行:n -= 1 (使得n變為2)-->yield n =2。這是x才得到了n=2這個值,并且被print了出來。依次類推,因此得到了序列10, 2, 1, 0

2016.12.11 更新:

注:以上代碼適用于python2。在python3中,generator的next()方法被改名為__next__()

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,505評論 6 533
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,556評論 3 418
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,463評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,009評論 1 312
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,778評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,218評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,281評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,436評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,969評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,795評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,993評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,537評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,229評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,659評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,917評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,687評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,990評論 2 374

推薦閱讀更多精彩內容