Python自定义迭代器中__iter__方法返回self的作用是什么?
为什么
__iter__方法要返回self? 在Python的迭代器协议中,有两个核心规则:
- 可迭代对象(iterable)必须实现
__iter__()方法,该方法必须返回一个迭代器对象 - 迭代器对象(iterator)必须实现
__next__()方法,用于生成下一个元素,无元素可生成时需抛出StopIteration异常
回到你的代码,MyNumbers类同时承担了两个角色:
- 它是可迭代对象:因为实现了
__iter__()方法 - 它也是迭代器:因为实现了
__next__()方法
所以在__iter__()里返回self,本质是把当前类实例作为迭代器返回给调用者(也就是iter(mynumbers)这行代码)。这样后续调用next(mynum)时,Python就能直接调用这个实例的__next__()方法来生成下一个数值。
另外,__iter__()里的self.n = 0会在每次调用iter()时重置计数器——比如你再次执行mynum = iter(mynumbers),计数器会回到0,后续next()调用会重新从1开始输出。
如果__iter__()不返回self,你就必须返回另一个实现了__next__()的对象,但这里的设计让类自身兼具可迭代和迭代器能力,返回self是最简洁合理的实现方式。
内容的提问来源于stack exchange,提问作者rowsen
相关产品推荐
相关产品推荐

