Python学习-39.Python中的生成器

先回顾列表解释

lista = range(10)
listb = [elem * elem for elem in lista]

那么listb就将会是0至9的二次方。

现在有这么一个需求，需要存储前10个斐波那契数到硬盘。

那么先写产生斐波那契数的函数：

def fib(max):
    n,a,b = 0,0,1
    while n < max:
        print(b)
        a,b = b,a + b
        n+=1

这样就会打印出前max个斐波那契数了。接着我们再修改一下。（因为我们需要的是存到硬盘里）

fiblist = []
def fib(max):
    n,a,b = 0,0,1
    while n < max:
        fiblist.append(b)
        a,b = b,a + b
        n+=1

fib(10)
print(fiblist)

接下来就可以操作fiblist保存到硬盘里了。

这样做可能我们操作前10个、前100个或许没有问题，但是假如某天业务需求要存十万个那就倒霉了。fiblist的长度得要十万，内存都撑不下了。这怎么办，直接修改fib函数，让其直接写入硬盘而不是追加到列表？但如果某天业务逻辑又改了怎么办，不存硬盘了，存数据库了。这显然不是好办法。

Python中有一个关键字，叫yield，就是可以解决这种问题的。

修改代码：

def fib(max):
    n,a,b = 0,0,1
    while n < max:
        yield b
        a,b = b,a + b
        n+=1

temp = fib(10)
print(temp)

现在，输出的则变成了<generator object fib at 0x0354FC10>

PS：at后面的0x部分可能有不同。

这是什么东西呢？这就是Python中的生成器。也就是一个可遍历对象。

既然可遍历，那么我们就可以使用for语句了

def fib(max):
    n,a,b = 0,0,1
    while n < max:
        yield b
        a,b = b,a + b
        n+=1

for i in fib(10000):
    #do something
    pass

现在该干什么就干什么吧，内存不会撑爆了。

另外，Python中的列表解释还能够快速改为生成器：

l = [elem * elem for elem in range(10)]

这是列表解释，而改为生成器：

g = (elem * elem for elem in range(10))

区别就只是将[]改为()

总结：

生成器就像是C#中的IEnumerable接口（这并不准确，准确是含GetEnumerator的方法）