(资料图片仅供参考)
在 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)
输出:
01234
在循环中,for
语句自动调用 iter()
函数获取迭代器,然后重复调用 __next__
方法获取下一个元素,直到发生 StopIteration
异常为止。
关键词: