Python 停止迭代
StopIteration
为了防止迭代无限进行,我们可以使用 StopIteration
语句。
在 __next__()
方法中,我们可以添加一个终止条件,如果迭代执行了指定次数,则抛出错误。
示例
停止 20 次迭代后
class MyNumbers
def __iter__(self)
self.a = 1
return self
def __next__(self)
if self.a <= 20
x = self.a
self.a += 1
return x
else
raise StopIteration
myclass = MyNumbers()
myiter = iter(myclass)
for x in myiter
print(x)
尝试一下 »