生成器
在python中,使用了yield的函數被稱為生成器(generator)。
生成器是一個返回迭代器的函數,只能用于迭代操作,生成器就是一個迭代器,
在調用生成器運行的過程中,每次遇到yield時函數會暫停并保存當前所有的運行信息,
并返回yield的值,并在下一次執行next()方法時從當前位置繼續執行。
yield的好處: 節省內存,只在需要時才生成一個。
迭代器也不占內存,for循環中用哪個取哪個
創建的兩種方法:
- (列表生成式)
- 使用yield關鍵字
用 yield 實現斐波那契數列:
#!/usr/bin/python3
import sys
def fibonacci(n): # 生成器函數 - 斐波那契
a, b, counter = 0, 1, 0
while True:
if (counter > n):
return
yield a
a, b = b, a + b
counter += 1
f = fibonacci(10) # f 是一個迭代器,由生成器返回生成
while True:
try:
print (next(f), end=" ")
except StopIteration:
sys.exit()
執行以上程序,輸出結果如下:
0 1 1 2 3 5 8 13 21 34 55