只知道使用迭代器能节省内存,但却不知道怎么使用才能节省内存。
下面就来回答这个问题。
首先创建一个list:
In [1]: a=[1,3,5,9,10]
其次,列表内每个元素+1,创建一个新列表
In [2]: a1 = [i+1 for i in a]
依次打印a1
中每个元素:
In [8]: for i in a1:
...: print(i)
上面操作等价于:
a1 = []
for i in a:
a1.append(i+1)
for i in a1:
print(i)
空间复杂度为 O(n),n为列表a内元素个数。
但是,使用迭代器实现上面的元素+1并打印的空间复杂度是多少呢?
ait = (i+1 for i in a) # 得到生成器,也是一种特殊的迭代器
for i in ait:
print(i)
上面操作等价于:
for i in a:
print(i+1)
不需要额外空间,所以使用迭代器加1并打印的空间复杂度为O(1).
结论:迭代器更加节省空间。
迭代器对象必须要实现两个方法__iter__
和__next__
,为了帮助大家理解,我们通过自定义一个迭代器加深对它的理解。
自定义一个迭代器,实现斐波那契数列:
#斐波那契数列
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项:
In [13]: for item in Fabs(10):
...: print(item,end=' ')
1 1 2 3 5 8 13 21 34 55
本文分享自 程序员郭震zhenguo 微信公众号,前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!