Python Остановить итерацию
Остановить итерацию
Чтобы итерация не продолжалась вечно, мы можем использовать
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)