首页 > 代码库 > Python学习-39.Python中的生成器
Python学习-39.Python中的生成器
先回顾列表解释
1 lista = range(10)2 listb = [elem * elem for elem in lista]
那么listb就将会是0至9的二次方。
现在有这么一个需求，需要存储前10个斐波那契数到硬盘。
那么先写产生斐波那契数的函数：
1 def fib(max):2 n,a,b = 0,0,13 while n < max:4 print(b)5 a,b = b,a + b6 n+=1
这样就会打印出前max个斐波那契数了。接着我们再修改一下。（因为我们需要的是存到硬盘里）
1 fiblist = [] 2 def fib(max): 3 n,a,b = 0,0,1 4 while n < max: 5 fiblist.append(b) 6 a,b = b,a + b 7 n+=1 8 9 fib(10)10 print(fiblist)
接下来就可以操作fiblist保存到硬盘里了。
这样做可能我们操作前10个、前100个或许没有问题，但是假如某天业务需求要存十万个那就倒霉了。fiblist的长度得要十万，内存都撑不下了。这怎么办，直接修改fib函数，让其直接写入硬盘而不是追加到列表？但如果某天业务逻辑又改了怎么办，不存硬盘了，存数据库了。这显然不是好办法。
Python中有一个关键字，叫yield，就是可以解决这种问题的。
修改代码：
1 def fib(max):2 n,a,b = 0,0,13 while n < max:4 yield b5 a,b = b,a + b6 n+=17 8 temp = fib(10)9 print(temp)
现在，输出的则变成了<generator object fib at 0x0354FC10>
PS：at后面的0x部分可能有不同。
这是什么东西呢？这就是Python中的生成器。也就是一个可遍历对象。
既然可遍历，那么我们就可以使用for语句了
1 def fib(max): 2 n,a,b = 0,0,1 3 while n < max: 4 yield b 5 a,b = b,a + b 6 n+=1 7 8 for i in fib(10000): 9 #do something10 pass
现在该干什么就干什么吧，内存不会撑爆了。
另外，Python中的列表解释还能够快速改为生成器：
1 l = [elem * elem for elem in range(10)]
这是列表解释，而改为生成器：
1 g = (elem * elem for elem in range(10))
区别就只是将[]改为()
总结：
生成器就像是C#中的IEnumerable接口（这并不准确，准确是含GetEnumerator的方法）
Python学习-39.Python中的生成器
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。