迭代器怎么节省内存呢?

迭代器怎么节省内存呢?今天番茄加速就来说一下。

只知道使用迭代器能节省内存，但却不知道怎么使用才能节省内存，下面就来回答这个问题。

首先创建一个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).

结论：迭代器更加节省空间!

如何自定义一个迭代器?

上面说过，迭代器对象必须要实现两个方法，为了更加具体，我们演示如何自定义一个迭代器。