Iterator 迭代器如何实现自定义遍历

要实现自定义遍历，您需要首先了解什么是迭代器（Iterator）。迭代器是一个对象，它允许您遍历一个聚合数据结构（如列表、集合或字典）中的元素，而无需暴露其内部表示。迭代器通过定义一个 __iter__() 方法和一个 __next__() 方法来实现遍历。

以下是一个简单的自定义迭代器示例，用于遍历一个列表：

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

    def __iter__(self):
        return self

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

# 使用自定义迭代器
my_list = [1, 2, 3, 4, 5]
custom_iterator = CustomIterator(my_list)

for item in custom_iterator:
    print(item)

在这个示例中，我们创建了一个名为 CustomIterator 的类，它接受一个列表作为输入。__iter__() 方法返回迭代器对象本身，而 __next__() 方法负责遍历列表并返回每个元素。当遍历完成时，__next__() 方法会引发 StopIteration 异常，通知循环结束。

您可以根据需要修改这个示例，以适应其他聚合数据结构或实现不同的遍历逻辑。