在 Python 中,迭代器是一个实现了 __iter__
和 __next__
方法的对象。__iter__
方法返回迭代器对象自身,而 __next__
方法返回下一个元素。换句话说,迭代器是一个可以逐个返回元素的对象。
下面是一个简单的迭代器示例,演示了如何实现 __iter__
和 __next__
方法:
class MyIterator:
def __init__(self, max_value):
self.max_value = max_value
self.current_value = 0
def __iter__(self):
return self
def __next__(self):
if self.current_value >= self.max_value:
raise StopIteration
value = self.current_value
self.current_value += 1
return value
该迭代器可以生成从 0 到 max_value - 1
的整数序列。__iter__
方法返回迭代器对象自身,而 __next__
方法返回下一个元素。当没有更多元素时,__next__
方法引发 StopIteration
异常,表示迭代已完成。
下面的示例演示了如何使用 MyIterator
类来迭代并打印从 0 到 4 的整数序列:
it = MyIterator(5)
for i in it:
print(i)
输出:
0
1
2
3
4
在循环中,for
语句自动调用 iter()
函数获取迭代器,然后重复调用 __next__
方法获取下一个元素,直到发生 StopIteration
异常为止。