在Python中,__iter__
方法是一个特殊的方法,它用于定义一个对象的迭代器行为。这个方法是面向对象编程中实现可迭代对象(即支持迭代的对象)的一部分。当你创建一个类,并且希望其实例能够在例如 for
循环中被迭代时,你就需要实现这个方法。
基本概念
- 可迭代对象:如果一个对象定义了
__iter__
方法,它就是可迭代的。该方法需要返回一个迭代器对象,该对象本身包含一个__next__
方法用于获取序列中的下一个元素。 - 迭代器:迭代器是实现了
__iter__
和__next__
方法的对象。__iter__
方法返回迭代器本身;__next__
方法返回序列中的下一个值。如果序列中没有更多元素,__next__
应该抛出一个StopIteration
异常。
实现 __iter__
实现 __iter__
方法的一般方式是让它返回一个对象,该对象有一个 __next__
方法。这个返回的对象可以是任何包含 __next__
方法的对象,包括实例本身。
示例:迭代一个自定义的序列类
假设我们想创建一个表示简单数列的类,这个类可以迭代从0开始到给定上限的整数。以下是如何实现这个类的一个例子: