Python中的yield到底是個什么鬼?
相信你已經不止一次在函數中看到關鍵詞yield,它起著什么作用?返回什么?和return又有著什么區別呢?這篇文章將會揭開yield的神秘面紗,并給出最淺顯易懂的例子。
yield關鍵字做了什么?
如果不太好理解yield,可以先把yield當作return的同胞兄弟來看,他們都在函數中使用,并履行著返回某種結果的職責。
這兩者的區別是:
有return的函數直接返回所有結果,程序終止不再運行,并銷毀局部變量;
而有yield的函數則返回一個可迭代的 generator(生成器)對象,你可以使用for循環或者調用next()方法遍歷生成器對象來提取結果。
什么是生成器呢?在 Python 中,使用了yield的函數被稱為生成器。有點套娃的感覺,但事實就是這樣,調用一個yield函數,就會返回一個生成器對象。
在調用生成器函數的過程中,每次遇到 yield 時函數會暫停并保存當前所有的運行信息(保留局部變量),返回yield的值, 并在下一次執行next()方法時從當前位置繼續運行,直到生成器被全部遍歷完。
先看一個簡單的例子:
上面函數返回一個生成器,現在對該生成器進行遍歷:
或者調用next()方法:
可以看到,simple_generator函數返回一個生成器,調用next()方法后,函數開始運行,遇到第一個yield關鍵字,返回生成的值(1),程序暫停;
第二次調用next()方法,代碼從上次暫停的位置開始執行,并遇到了第二個yield關鍵字,再返回生成的值(2),程序暫停;
第三次調用也是如此,返回生成的值(3),生成器耗盡,程序終止;
到這里你可能就明白yield和return的關系和區別了,帶yield的函數是一個生成器,這個生成器有一個方法就是next,next就相當于“下一步”生成哪個數,這一次的next開始的地方是接著上一次的next停止的地方執行的。
所以調用next的時候,生成器并不會從函數的開始執行,只是接著上一步停止的地方開始,然后遇到yield后,return出要生成的數,此步就結束。
有個經典的例子就是使用yield生成斐波那契數列:
- def fab(max):
- n, a, b = 0, 0, 1
- while n < max:
- yield b # 使用 yield
- a, bb = b, a + b
- nn = n + 1
- for n in fab(5):
- print n
生成器有哪些作用?
如果想具體化數據的形式,通常會將數據存儲在一個列表中。但這樣做,列表的內容將占用有形內存。列表越大,占用的內存資源就越多。
但是,如果數據集有某種邏輯,就不必存儲在一個列表中,只需編寫一個生成器,它將在需要時生成這些值,基本不占用內存。