【数据结构】【python】迭代器及其优点

迭代器

（1）定义

迭代器是一个可以记住遍历的位置的对象。迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。迭代器不仅可以用于已知的数据结构，还可以自己构造迭代器。例如，ist 是可迭代对象，dict 也是可迭代对象。

当运行如下代码时：

example = [1,2,3,4,5]
for i in example:
	print(i)

实际上，list是一个可迭代对象，其内部使用python内置函数iter()调用遍历对象的__iter__()获得一个迭代器，之后再循环对这个迭代器使用next()调用迭代器对象的__next__()。iter()只会被调用一次，而__next__()会被调用 n 次。

（2）优点

迭代器的优势有以下几点：

（1）降低内存消耗：迭代器可以逐个处理数据，不需要一次性将整个数据集加载到内存中，因此可以有效降低内存消耗。这点是源自于它的懒加载的特性，即每次只加载需要使用的内容。

（2）方便迭代：迭代器可以在不同的场景下使用，因此可以增加代码的灵活性。迭代器有两个基本的方法：iter() 和 next()。
例如对于list结构，可以使用next去访问。实际上

>>> list=[1,2,3,4]
>>> it = iter(list)    # 创建迭代器对象
>>> print (next(it))   # 输出迭代器的下一个元素
1
>>> print (next(it))
2

或者自己定义一个迭代器：

class test(object):
    def __init__(self, cnt):
        self.val = 0
        self.cnt = cnt

    def __iter__(self):
        return self

    def __next__(self):
        if self.cnt > 0:
            self.cnt -= 1
            self.val += 1
            return self.val
        else:
            raise StopIteration

if __name__ == '__main__':
    res = test(5)
    for i in res:
    	print(i)

运行代码得到的结果是：1，2，3，4，5.因此迭代器不止运用在常见的数据结构中，也可以是自己定义的类。总而言之，需要是可迭代对象。