深入理解 Python 迭代器

Python 迭代器是一种强大的工具，它允许我们以一种统一的方式遍历各种数据结构。在 Python 中，迭代器是实现了迭代器协议的对象，它提供了两个方法：__iter__() 和 __next__()。

__iter__() 方法用于返回迭代器对象本身，而 __next__() 方法用于返回迭代器中的下一个元素。当迭代器中没有更多元素时，__next__() 方法会抛出 StopIteration 异常。

使用迭代器的主要优点是它提供了一种简洁、统一的方式来遍历各种数据结构，而无需关心数据结构的具体实现细节。此外，迭代器还可以节省内存，因为它只在需要时才计算下一个元素，而不是一次性将所有元素都加载到内存中。Python 内置了许多可迭代对象，如列表、元组、字典、字符串等。我们可以通过调用这些对象的 iter() 方法来获取它们的迭代器。例如，对于列表 my_list，我们可以使用以下代码获取其迭代器：

my_iterator = iter(my_list)

然后，我们可以使用 next() 函数来获取迭代器中的下一个元素：

next_element = next(my_iterator)

除了内置的可迭代对象，我们还可以自定义迭代器。自定义迭代器需要实现 __iter__() 和 __next__() 方法。下面是一个简单的示例，展示了如何创建一个自定义迭代器：

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

    def __iter__(self):
        return self

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

在上面的示例中，我们创建了一个名为 MyIterator 的类，它实现了迭代器协议。__init__() 方法用于初始化迭代器的状态，__iter__() 方法返回迭代器对象本身，__next__() 方法用于返回迭代器中的下一个元素。使用自定义迭代器的方式与使用内置迭代器相同。我们可以创建一个 MyIterator 对象，并使用 next() 函数来获取其元素：

my_iterator = MyIterator([1, 2, 3])
next_element = next(my_iterator)