正文

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

IP云
IP云V管理员/4 阅读
11/30
提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

先回顾列表解释 

 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的方法）

相关阅读

目录[+]