生成器
为什么要使用生成器
简单一句话:节省内存和计算,做到“即用即取”
为什么可以节省内存?
因为生成器可以保存函数或列表生成器的算法,而不是将结果一次性全部计算出来,而是在每次调用时,产生一个值。
生成器的二种常用形式
1.列表生成器
一个列表生成式:
[x*x for x in range(1,5)]
结果:
此时将列表生成式的方括号改为圆括号即可变为生成器:
(x*x for x in range(1,5))
结果:
返回了一个生成器对象(generator object)
用一个变量a,接收后,进行next迭代:
2.yield关键字
在一个函数中将return改为yield关键字,即可将函数转变成一个生成器。
经典的Fibonacci问题(斐波那契数列)
1,1,2,3,5,8,13……
从第三项起,每一个数字都是前两个数之和
采用非递归的方式进行编写:
def fibonacciNums(n):
a, b = 0, 1
for i in range(n):
print(b)
a, b = b, a+b
fibonacciNums(7)
改为生成器:
def fibonacciNums(n):
a, b = 0, 1
for i in range(n):
yield b
a, b = b, a+b
res = fibonacciNums(7)
print(res)
调用后,返回了一个生成器对象。
进行next迭代,才会产生数据:
def fibonacciNums(n):
a, b = 0, 1
for i in range(n):
yield b
a, b = b, a+b
res = fibonacciNums(7)
print(next(res))
print(next(res))
print(next(res))
print(next(res))
print(next(res))
结果: