Python中的yield到底是個什么鬼？

點擊上方"藍字"關注我們

相信你已經(jīng)不止一次在函數(shù)中看到關鍵詞yield，它起著什么作用？返回什么？和return又有著什么區(qū)別呢？這篇文章將會揭開yield的神秘面紗，并給出最淺顯易懂的例子。

yield關鍵字做了什么？

如果不太好理解yield，可以先把yield當作return的同胞兄弟來看，他們都在函數(shù)中使用，并履行著返回某種結果的職責。

這兩者的區(qū)別是：

有return的函數(shù)直接返回所有結果，程序終止不再運行，并銷毀局部變量；

而有yield的函數(shù)則返回一個可迭代的 generator（生成器）對象，你可以使用for循環(huán)或者調(diào)用next()方法遍歷生成器對象來提取結果。

什么是生成器呢？在 Python 中，使用了yield的函數(shù)被稱為生成器。有點套娃的感覺，但事實就是這樣，調(diào)用一個yield函數(shù)，就會返回一個生成器對象。

在調(diào)用生成器函數(shù)的過程中，每次遇到 yield 時函數(shù)會暫停并保存當前所有的運行信息（保留局部變量），返回yield的值, 并在下一次執(zhí)行next()方法時從當前位置繼續(xù)運行，直到生成器被全部遍歷完。

先看一個簡單的例子：

上面函數(shù)返回一個生成器，現(xiàn)在對該生成器進行遍歷：

或者調(diào)用next()方法：

可以看到，simple_generator函數(shù)返回一個生成器，調(diào)用next()方法后，函數(shù)開始運行，遇到第一個yield關鍵字，返回生成的值（1），程序暫停；

第二次調(diào)用next()方法，代碼從上次暫停的位置開始執(zhí)行，并遇到了第二個yield關鍵字，再返回生成的值（2），程序暫停；

第三次調(diào)用也是如此，返回生成的值（3），生成器耗盡，程序終止；

到這里你可能就明白yield和return的關系和區(qū)別了，帶yield的函數(shù)是一個生成器，這個生成器有一個方法就是next，next就相當于“下一步”生成哪個數(shù)，這一次的next開始的地方是接著上一次的next停止的地方執(zhí)行的。

所以調(diào)用next的時候，生成器并不會從函數(shù)的開始執(zhí)行，只是接著上一步停止的地方開始，然后遇到yield后，return出要生成的數(shù)，此步就結束。

有個經(jīng)典的例子就是使用yield生成斐波那契數(shù)列：

def?fab(max):?
????n,?a,?b?=?0,?0,?1?
????while?n?????????yield?b??????#?使用?yield
?
????????a,?b?=?b,?a?+?b?
????????n?=?n?+?1
?
for?n?in?fab(5):?
????print?n

生成器有哪些作用？

如果想具體化數(shù)據(jù)的形式，通常會將數(shù)據(jù)存儲在一個列表中。但這樣做，列表的內(nèi)容將占用有形內(nèi)存。列表越大，占用的內(nèi)存資源就越多。

但是，如果數(shù)據(jù)集有某種邏輯，就不必存儲在一個列表中，只需編寫一個生成器，它將在需要時生成這些值，基本不占用內(nèi)存。

Python大數(shù)據(jù)分析

data creates?value