如何在for循环break后重置自定义可迭代类ObjectSet的指针？

你的问题核心在于当前的ObjectSet把自身作为迭代器（__iter__返回self），这就导致迭代过程中的pointer状态会被保留下来——当你用break终止循环时，pointer停在了断点位置，下一次迭代就从这里继续了。而列表的迭代逻辑是每次for循环都会生成一个全新的迭代器，所以每次都能从头开始遍历。

优雅的修复方案：让__iter__返回独立的迭代器实例 #

我们可以把迭代器的状态（也就是pointer）从ObjectSet类中分离出来，让__iter__返回一个专门的迭代器对象，这样每次for循环都会创建一个新的迭代器，彼此的状态互不干扰。

修改后的代码如下：

class ObjectSet:
    def __init__(self, objects=None):
        self.objects = objects if objects is not None else []
    
    def __iter__(self):
        # 返回一个独立的迭代器对象，拥有自己的pointer
        return ObjectSetIterator(self.objects)
    
    def __len__(self):
        return len(self.objects)

class ObjectSetIterator:
    def __init__(self, objects):
        self.objects = objects
        self.pointer = 0
    
    def __next__(self):
        if self.pointer >= len(self.objects):
            raise StopIteration
        current_obj = self.objects[self.pointer]
        self.pointer += 1
        return current_obj

为什么这样有效？ #

• 每次调用for obj in objs时，Python会自动调用objs.__iter__()，这里我们返回了一个新的ObjectSetIterator实例，每个实例都有自己的pointer，初始值都是0。
• 不管上一次循环是正常结束还是用break终止，新的迭代器都不受影响，都会从第一个元素开始遍历。

测试你的代码 #

现在运行你给出的测试代码：

objs = ObjectSet([1,2,3])
for obj in objs:
    if obj == 2:
        break
for obj in objs:
    print(obj)

会输出：

1
2
3

完全符合你的预期。

另一种更简洁的写法（用生成器） #

如果你不想额外写一个迭代器类，还可以用生成器函数来实现__iter__，代码会更紧凑：

class ObjectSet:
    def __init__(self, objects=None):
        self.objects = objects if objects is not None else []
    
    def __iter__(self):
        # 生成器自动维护迭代状态，每次调用都是新的生成器
        for obj in self.objects:
            yield obj
    
    def __len__(self):
        return len(self.objects)

这种方式同样能达到效果，生成器本身就是天然的迭代器，每次调用__iter__都会生成一个新的生成器实例，状态完全独立。

内容的提问来源于stack exchange，提问作者Anton Fahlgren