Python 自定义可迭代对象的正确姿势_网络技术_宿州市腾雀网络科技有限公司

__iter__ 必须返回迭代器而非数据，因Python迭代机制只认迭代器协议（含__next__且耗尽时抛StopIteration）；直接返回列表等虽可迭代，但丧失控制权、无法干预next()行为、多次遍历状态共享。

为什么 `iter` 必须返回迭代器，而不是直接返回数据？

因为 Python 的 for 循环、list()、next() 等机制只认「迭代器协议」：对象要有 __next__ 方法，且在耗尽时抛出 StopIteration。如果 __iter__ 直接返回列表或生成器表达式（比如 return self.data），而 self.data 是 list，那它确实可迭代——但这是靠 list 自己实现了迭代器协议；你的类只是“借壳”，没真正控制迭代逻辑。

正确做法是让 __iter__ 返回一个满足协议的对象：

返回一个自定义迭代器类的实例（推荐，逻辑清晰、可维护）
返回生成器函数（即含 yield 的函数）的调用结果（简洁，但状态封装弱）
避免返回 iter(self.data) 除非你明确只需要代理行为且不扩展逻辑

示例中常见错误：def __iter__(self): return self.data —— 这会让多次遍历共享内部状态（比如 list 的索引不可控），也失去对 next() 行为的干预能力。

自定义迭代器类里，`next` 怎么写才不出错？

核心就两点：有明确的终止条件 + 每次调用返回下一个值。最容易踩的坑是忘记维护游标、越界不抛异常、或在耗尽后继续返回值。

典型结构：

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

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

必须显式检查边界，不能依赖 try/except IndexError —— 迭代器协议要求抛 StopIteration，不是 IndexError
不要在 __next__ 里重置 self.index，否则会导致无限循环
如果数据支持动态修改（如边迭代边增删），需额外考虑一致性，通常建议迭代期间禁止修改