Python中的迭代器

一、介绍

在Python中，迭代器是一种访问集合元素的方式，可以用于遍历数据集中的元素，而不需要事先知道集合的大小。迭代器可以被用于循环语句中，例如for循环，来遍历集合中的每个元素。

Python中的迭代器是一个实现了迭代器协议（Iterator Protocol）的对象。迭代器协议包含了两个方法：

• __iter__()：返回迭代器对象本身。
• __next__()：next()方法返回容器中的下一个元素。当容器中的所有元素都被访问完毕后，next()方法会引发StopIteration异常，从而结束迭代。

迭代器的优点是可以按需生成元素，避免一次性加载所有元素造成的内存浪费；同时，迭代器可以实现惰性求值，避免不必要的计算，提高程序效率。

二、实例

# -*- coding: UTF-8 -*-

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = -1

    def __iter__(self):
        return self

    def __next__(self):
        self.index += 1
        if self.index < len(self.data):
            return self.data[self.index]
        raise StopIteration

my_list = [1, 2, 3, 4, 5]
iterator = MyIterator(my_list)

for item in iterator:
    print(item)