Python迭代器的__iter__和__next__详细教程

在 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 异常为止。

你可能感兴趣的:(python,迭代器模式)