python 迭代器

# 示例代码：Python 迭代器

# 定义一个简单的迭代器类
class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __iter__(self):
        # 返回迭代器对象本身
        return self

    def __next__(self):
        # 如果索引超出范围，触发 StopIteration 异常
        if self.index >= len(self.data):
            raise StopIteration
        # 获取当前索引对应的元素，并将索引加1
        result = self.data[self.index]
        self.index += 1
        return result

# 创建一个列表并使用自定义迭代器遍历它
my_list = [1, 2, 3, 4, 5]
iterator = MyIterator(my_list)

# 使用 for 循环遍历迭代器
for item in iterator:
    print(item)

# 解释说明：
# 1. `__iter__` 方法返回迭代器对象本身。
# 2. `__next__` 方法返回下一个元素，如果没有更多元素，则抛出 StopIteration 异常。
# 3. 在 for 循环中，迭代器会自动调用 `__next__` 方法直到遇到 StopIteration 异常。