生成器

一、什麼是生成器

接觸了列表生成式,我們可以用列表生成式創建一個列表。但受內存限制,列表的容量不可能無限大,而且如果一個包含很多元素的列表而我們只訪問其中幾個,這樣就會白白浪費很多空間。所以我們有這樣一個想法,我們只保存得出列表的算法,再根據需要去計算需要的元素,在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語句處繼續執行。

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章