如何循环一个类实例？

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