如何让类支持迭代协议？

要让一个类支持迭代协议，我们需要实现__iter__和__next__方法。1) 在__iter__方法中初始化迭代状态并返回迭代器对象。2) 在__next__方法中定义每次迭代的逻辑，并在迭代结束时抛出stopiteration异常。这使得类可以被用于for循环和其他迭代场景，增强了代码的灵活性和可读性。

要让一个类支持迭代协议，我们需要实现 __iter__ 和 __next__ 方法，使得这个类可以被用在 for 循环和其他需要迭代的地方。让我详细解释一下这个过程，并分享一些我自己的经验和思考。

当我第一次接触到迭代协议的时候，我觉得这是一个非常巧妙的设计，它使得 Python 的代码可以更加简洁和优雅。通过实现迭代协议，我们可以让自定义的类和内置的可迭代对象一样，轻松地被遍历。这不仅提高了代码的可读性，也增强了语言的灵活性。

让我们从一个简单的例子开始，来看如何实现一个自定义的迭代器。假设我们想要创建一个类 Countdown，它可以从一个给定的数字开始倒数：

class Countdown:    def __init__(self, start):        self.start = start<pre class='brush:php;toolbar:false;'>def __iter__(self):    self.n = self.start    return selfdef __next__(self):    if self.n < 0:        raise StopIteration    result = self.n    self.n -= 1    return result

登录后复制