【Python】Python中类的__iter__()方法

在Python中，__iter__ 方法是一个特殊的方法，它用于定义一个对象的迭代器行为。这个方法是面向对象编程中实现可迭代对象（即支持迭代的对象）的一部分。当你创建一个类，并且希望其实例能够在例如 for 循环中被迭代时，你就需要实现这个方法。

基本概念

1. 可迭代对象：如果一个对象定义了 __iter__ 方法，它就是可迭代的。该方法需要返回一个迭代器对象，该对象本身包含一个 __next__ 方法用于获取序列中的下一个元素。
2. 迭代器：迭代器是实现了 __iter__ 和 __next__ 方法的对象。__iter__ 方法返回迭代器本身；__next__ 方法返回序列中的下一个值。如果序列中没有更多元素，__next__ 应该抛出一个 StopIteration 异常。

实现 __iter__

实现 __iter__ 方法的一般方式是让它返回一个对象，该对象有一个 __next__ 方法。这个返回的对象可以是任何包含 __next__ 方法的对象，包括实例本身。

示例：迭代一个自定义的序列类

假设我们想创建一个表示简单数列的类，这个类可以迭代从0开始到给定上限的整数。以下是如何实现这个类的一个例子：

class Counter:
    def __init__(self, low, high):
        self.current = low
        self.high = high

    def __iter__(self):
        # 返回迭代器对象，这里是类本身
        return self

    def __next__(self):
        # 如果当前值小于上限，返回当前值并递增
        if self.current < self.high:
            num = self.current
            self.current += 1
            return num
        # 如果达到或超过上限，抛出 StopIteration
        raise StopIteration

这里，Counter 类通过实现 __iter__ 和 __next__ 方法成为了一个迭代器。你可以用一个 for 循环来迭代这个类的实例：

counter = Counter(1, 4)
for num in counter:
    print(num)  # 输出 1, 2, 3

使用生成器简化

对于简单的迭代逻辑，你可以使用生成器来自动处理 __iter__ 和 __next__ 的实现。这通过在一个方法中使用 yield 语句来实现：

class Counter:
    def __init__(self, low, high):
        self.low = low
        self.high = high

    def __iter__(self):
        current = self.low
        while current < self.high:
            yield current
            current += 1

这里，__iter__ 方法是一个生成器，它自动创建了一个迭代器对象。

总结

通过定义 __iter__ 方法，你可以让你的类支持迭代，这在处理序列数据、自定义数据结构或需要逐个元素处理的情况下特别有用。此外，使用生成器可以简化迭代器的创建和管理。