前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >迭代器怎么就节省内存了?

迭代器怎么就节省内存了?

作者头像
double
发布2020-10-14 17:43:32
1.3K0
发布2020-10-14 17:43:32
举报
文章被收录于专栏:算法channel

迭代器怎么节省内存了?

只知道使用迭代器能节省内存,但却不知道怎么使用才能节省内存。

下面就来回答这个问题。

首先创建一个list:

代码语言:javascript
复制
In [1]: a=[1,3,5,9,10]

其次,列表内每个元素+1,创建一个新列表

代码语言:javascript
复制
In [2]: a1 = [i+1 for i in a]  

依次打印a1中每个元素:

代码语言:javascript
复制
In [8]: for i in a1: 
   ...:     print(i) 

上面操作等价于:

代码语言:javascript
复制
a1 = []
for i in a:
  a1.append(i+1)
for i in a1:
  print(i)

空间复杂度为 O(n),n为列表a内元素个数。

但是,使用迭代器实现上面的元素+1并打印的空间复杂度是多少呢?

代码语言:javascript
复制
ait = (i+1 for i in a) # 得到生成器,也是一种特殊的迭代器
for i in ait:
  print(i)

上面操作等价于:

代码语言:javascript
复制
for i in a:
  print(i+1)

不需要额外空间,所以使用迭代器加1并打印的空间复杂度为O(1).

结论:迭代器更加节省空间。

自定义迭代器案例

迭代器对象必须要实现两个方法__iter____next__,为了帮助大家理解,我们通过自定义一个迭代器加深对它的理解。

自定义一个迭代器,实现斐波那契数列:

代码语言:javascript
复制
#斐波那契数列
class Fabs():
    def __init__(self,max):
        self.max=max
        self.n,self.a,self.b=0,0,1
    
    #定义__iter__方法
    def __iter__(self):
        return self
        
    #定义__next__方法
    def __next__(self):
        if self.n<self.max:
            tmp=self.b
            self.a,self.b=self.b,self.a+self.b
            self.n+=1
            return tmp
        raise StopIteration

使用这个迭代器,打印斐波那契数列前10项:

代码语言:javascript
复制
In [13]: for item in Fabs(10): 
    ...:     print(item,end=' ')                                                                    
1 1 2 3 5 8 13 21 34 55 
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2020-10-06,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 程序员郭震zhenguo 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 迭代器怎么节省内存了?
  • 自定义迭代器案例
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档