Python迭代器协议怎样避免死循环

在Python中，迭代器协议要求实现两个方法：__iter__() 和 __next__()。为了避免死循环，我们需要确保在每次调用 __next__() 方法时，都能返回一个有效的元素，直到没有更多元素可以返回为止。

以下是一个简单的示例，展示了如何实现一个避免死循环的迭代器：

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

# 使用示例
iterator = MyIterator(0, 5)
for num in iterator:
    print(num)

在这个例子中，MyIterator 类实现了迭代器协议。__iter__() 方法返回迭代器对象本身，而 __next__() 方法在每次调用时检查当前值是否在指定范围内。如果在范围内，则返回当前值并将迭代器向前移动一个位置。当迭代器到达范围的末尾时，__next__() 方法会引发 StopIteration 异常，从而结束迭代。

这个实现可以避免死循环，因为每次调用 __next__() 方法时，我们都会检查是否还有元素可以返回。如果没有，我们会引发 StopIteration 异常，而不是继续执行循环。