生成器
在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