1. 程式人生 > >python裏的生成器--yield

python裏的生成器--yield

python叠代器 python生成器 yield理解 快速理解yield

Python的生成器是個很強大的東西,特別是在python3.0版本以後。以最簡單的方式讓大家快速理解生成器。

1、正常的寫法
來看個例子,比如輸出一個自定義長度的列表一般這麽寫:
技術分享圖片
這裏傳入的參數時10,所以會得到一個包含10個元素的列表:
技術分享圖片
那當我傳入的是10W的時候,那生成的這個列表就很大了,也占內存,運行腳本也占cpu。

2、改良後寫法
改良一下代碼,把他寫成一個叠代的類:
技術分享圖片
這裏面self.b就記錄了每次執行next方法的位置,知道每次是第幾次執行next方法,所以執行保證了每次輸出的是期望的值,其實這就是叠代了,每運行一次函數都被記錄已運行的狀態。當被調用的時候才返回值,否則就處於等待被調用的狀態
運行結果:

技術分享圖片
所以這改良後的代碼就解決了當你輸入10W的時候占用資源的問題,因為輸入10W後,只要當調用next函數的時候才返回值,不是一次返回一個那麽大的列表出來。
3、生成器
那麽第二步中的代碼跟第一步比起來又太多了感覺,那麽生成器就來了

再改良代碼:
技術分享圖片
只需改下第一步中的代碼a.append(n)為yield n,這就是一個生成器了,然後通過for語句來調用生成器的值。

任何一個帶有yield語句的函數都是生成器,當你直接調用這個函數時,內部的代碼是不會被執行的,只有調用yield裏面的next函數才會去執行代碼,for循環也就是會自動去調用這個next函數來輸出值。

可以理解為一個函數被yield中斷了,下載再次調用時繼續從上一次中斷的位置繼續執行代碼並返回值。

講的比較簡單,不知道大家理解了沒有。

python裏的生成器--yield