一、什麼是生成器
接觸了列表生成式,我們可以用列表生成式創建一個列表。但受內存限制,列表的容量不可能無限大,而且如果一個包含很多元素的列表而我們只訪問其中幾個,這樣就會白白浪費很多空間。所以我們有這樣一個想法,我們只保存得出列表的算法,再根據需要去計算需要的元素,在python中,這種一邊循環一邊計算的機制我們就稱爲生成器(generator)
二、創建生成器的方法
1.只需要把列表生成式的[]換成()即可以創建一個generator
對於生成器我們都可以用next()函數一個個打印,所以可以用next()函數的就是生成器。
當計算到最後一個元素,沒有更多元素時就會拋出StopIteration異常
但是使用next(g)去計算太過麻煩,使用我們也經常使用for循環
2.如果推算算法太複雜,用類似列表生成式的for循環沒辦法實現時,就可以用函數實現
著名的斐波拉契數列(Fibonacci),除第一個和第二個數外,任意一個數都可由前兩個數相加得到:
1, 1, 2, 3, 5, 8, 13, 21, 34, ...
斐波拉契數列用列表生成式寫不出來,但是,用函數把它打印出來卻很容易:
仔細觀察,可以看出,fib
函數實際上是定義了斐波拉契數列的推算規則,可以從第一個元素開始,推算出後續任意的元素,這種邏輯其實非常類似generator。
也就是說,上面的函數和generator僅一步之遙。要把fib
函數變成generator,只需要把print(b)
改爲yield b
就可以了:
這裏,最難理解的就是generator和函數的執行流程不一樣。函數是順序執行,遇到return
語句或者最後一行函數語句就返回。而變成generator的函數,在每次調用next()
的時候執行,遇到yield
語句返回,再次執行時從上次返回的yield
語句處繼續執行。