Python迭代器协议如何工作

Python 迭代器协议是一个定义了迭代器对象应遵循的规范

1. __iter__() 方法：这个方法返回迭代器对象本身。在 Python 中，可以使用 for 循环遍历一个可迭代对象，这是因为内置函数 iter() 会调用对象的 __iter__() 方法来获取迭代器。

class MyIterable:
    def __iter__(self):
        return self

2. __next__() 方法：这个方法返回迭代器的下一个值。当没有更多元素时，这个方法应该抛出一个 StopIteration 异常。在使用 for 循环遍历一个可迭代对象时，Python 会自动调用此方法以获取下一个元素。

class MyIterator:
    def __init__(self, start, end):
        self.start = start
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        if self.start >= self.end:
            raise StopIteration
        else:
            self.start += 1
            return self.start - 1

在这个例子中，MyIterator 类实现了迭代器协议，因为它定义了 __iter__() 和 __next__() 方法。这使得我们可以使用 for 循环遍历这个迭代器对象：

my_iterator = MyIterator(0, 5)
for number in my_iterator:
    print(number)

输出：

0
1
2
3
4

总之，Python 迭代器协议通过定义 __iter__() 和 __next__() 方法，使得对象可以被用于 for 循环，从而简化了遍历操作。