python實現斐波拉契數列
阿新 • • 發佈:2018-12-26
第一種
# 斐波拉契數列
def fib(max):
n, a, b = 0, 1, 1
while n < max:
print(b)
a, b = b, a+b
n = n+1
return 'done'
fib(5)
結果為
1
1
2
3
5
第二改成生成器模式
仔細觀察,可以看出,fib函式實際上是定義了斐波拉契數列的推算規則,可以從第一個元素開始,推算出後續任意的元素,這種邏輯其實非常類似generator。
也就是說,上面的函式和generator僅一步之遙。要把fib函式變成generator,只需要把print(b)改為yield b就可以了:
def fib(max):
n, a, b = 0, 1, 1
while n < max:
yield b
a, b = b, a+b
n = n+1
return 'done'
f= fib(5)
print(f)
for i in f:
print(i)
結果為
<generator object fib at 0x00000170CFFD6A98>
1
2
3
5
8