python必会之生成器

生成器

为什么要使用生成器

简单一句话：节省内存和计算，做到“即用即取”
为什么可以节省内存？
因为生成器可以保存函数或列表生成器的算法，而不是将结果一次性全部计算出来，而是在每次调用时，产生一个值。

生成器的二种常用形式

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))

结果：