创建Python可迭代对象时多循环失效问题求助

你遇到的问题根源在于：你的类同时扮演了「可迭代对象」和「迭代器」的角色。当__iter__返回self时，整个实例的迭代状态（也就是self.n）是全局共享的——第一次内层循环跑完后，self.n已经走到了终止条件，后续再循环时直接触发StopIteration，导致外层循环只能执行一次，内层循环也无法从头开始。

为什么原代码会输出1 2 1 3 1 4？ #

初始时self.n = 0，外层循环第一次取i：

• 调用__next__，0 < 4成立，self.n变成1，返回1（所以i=1）。
  然后进入内层循环：
• 第一次j：1 <4成立，self.n变成2，返回2→输出1 2。
• 第二次j：2 <4成立，self.n变成3，返回3→输出1 3。
• 第三次j：3 <4成立，self.n变成4，返回4→输出1 4。
• 第四次调用__next__：4 <4不成立，触发StopIteration，内层循环结束。

接下来外层循环尝试取下一个i：

• 调用__next__，此时self.n=4，依然不满足4<4，直接触发StopIteration，外层循环终止。
  所以最终只得到了三次输出，和你的预期不符。

解决方案：让每次迭代都拥有独立的状态 #

有三种常见的修复方式，你可以根据需求选择：

1. 在__iter__中重置状态（简单但有并发风险）

如果你的类实例不会被多个迭代同时遍历，最简单的方法是每次调用__iter__时重置计数器：

class test():
    def __init__(self):
        self.n = 0
    def __iter__(self):
        self.n = 0  # 每次迭代前重置计数器
        return self
    def __next__(self):
        if self.n < len(self)-1:
            self.n += 1
            return self.n
        else:
            raise StopIteration
    def __len__(self):
        return 5

测试你的循环：

test_instance = test()  # 注意变量名不要和类名重复！
for i in test_instance:
    for j in test_instance:
        print(i,j)

此时会输出你预期的1 1 1 2 1 3 1 4 2 1 2 2 ...，因为每次进入循环都会重置self.n。

2. 分离可迭代对象和迭代器（更健壮）

如果需要支持多个迭代同时进行（比如两个循环遍历同一个实例），推荐把迭代逻辑拆成独立的迭代器类：

class test():
    def __init__(self):
        self.length = 5
    def __iter__(self):
        # 每次迭代返回一个新的迭代器实例，状态独立
        return TestIterator(self.length)
    def __len__(self):
        return self.length

class TestIterator():
    def __init__(self, length):
        self.n = 0
        self.length = length
    def __next__(self):
        if self.n < self.length - 1:
            self.n += 1
            return self.n
        else:
            raise StopIteration
    def __iter__(self):
        return self

这种方式下，每个循环都会创建一个新的TestIterator，互相之间的状态完全独立，不会干扰。

3. 使用生成器实现（最简洁）

Python的生成器天然支持独立迭代状态，用yield可以大幅简化代码：

class test():
    def __init__(self):
        self.length = 5
    def __iter__(self):
        n = 0
        while n < self.length - 1:
            n += 1
            yield n
    def __len__(self):
        return self.length

生成器在每次调用__iter__时都会从头执行，自动维护独立的迭代状态，代码量最少，也是Python中最常用的实现方式。

额外提示：避免类名和变量名重复 #

你原代码里把实例命名为test，和类名test重名了，这会导致后续无法再创建新的类实例，建议改成test_instance这类名称。

内容的提问来源于stack exchange，提问作者Diego Aguayo