如何循环一个类实例？

Question

我被问到这个问题。鉴于代码：

class MockList:
    ...code_here...

for i in MockList():
    print(i)

for循环的预期结果：

1 2 3 4 5

我该怎么做？

Answer 1

如果您实施__iter__()方法，则可以执行此功能：

代码：

class MockList:

    def __iter__(self):
        return iter(range(1, 6))

测试代码：

for i in MockList():
    print(i)

结果：

1
2
3
4
5