在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
异常,而不是继续执行循环。