您可能聽說過,帶有 yield 的函數在 Python 中被稱之為 generator(產生器),何謂 generator ?
我們先拋開 generator,以一個常見的程式設計題目來展示 yield 的概念。
如何生成斐波那契數列
斐波那契(Fibonacci)數列是一個非常簡單的遞迴數列,除第一個和第二個數外,任意一個數都可由前兩個數相加得到。 用電腦程式輸出斐波那契數列的前 N 個數是一個非常簡單的問題,許多初學者都可以輕易寫出如下函數:
清單 1. 簡單輸出斐波那契數列前 N 個數
def fab(max): n, a, b = 0, 0, 1 while n < max: print b a, b = b, a + b n = n + 1
執行 fab(5),我們可以得到如下輸出:
>>> fab(5) 1 1 2 3 5
結果沒有問題,但有經驗的開發者會指出,直接在 fab 函數中用 print 列印數位會導致該函數可複用性較差,因為 fab 函數返回 None,其他函數無法獲得該函數生成的數列。
要提高 fab 函數的可複用性,最好不要直接列印出數列,而是返回一個 List。 以下是 fab 函數改寫後的第二個版本:
清單 2. 輸出斐波那契數列前 N 個數第二版
def fab(max): n, a, b = 0, 0, 1 L = [] while n < max: L.append(b) a, b = b, a + b n = n + 1 return L
可以使用如下方式列印出 fab 函數返回的 List:
>>> for n in fab(5): ... print n ... 1 1 2 3 5
改寫後的 fab 函數通過返回 List 能滿足複用性的要求,但是更有經驗的開發者會指出,該函數在運行中佔用的記憶體會隨著參數 max 的增大而增大,如果要控制HTTP://www.aliyun.com/zixun/ aggregation/17969.html">記憶體佔用,最好不要用 List