Python 3: преимущества и недостатки языка Python интерпретируемый язык программирования

1
2
3
4
5
Как уже было сказано, объекты, элементы которых можно перебирать в цикле for, содержат в себе объект итератор, для того, чтобы его получить необходимо использовать функцию iter(), а для извлечения следующего элемента из итератора – функцию next().
>>> itr = iter(num_list)
>>> print(next(itr))
1
>>> print(next(itr))
2
>>> print(next(itr))
3
>>> print(next(itr))
4
>>> print(next(itr))
5
>>> print(next(itr))
Traceback (most recent call last):
File "
", line 1, in
print(next(itr))
StopIteration
Как видно из приведенного выше примера вызов функции next(itr) каждый раз возвращает следующий элемент из списка, а когда эти элементы заканчиваются, генерируется исключение StopIteration.

Создание собственных итераторов

Если нужно обойти элементы внутри объекта вашего собственного класса, необходимо построить свой итератор. Создадим класс, объект которого будет итератором, выдающим определенное количество единиц, которое пользователь задает при создании объекта. Такой класс будет содержать конструктор, принимающий на вход количество единиц и метод __next__(), без него экземпляры данного класса не будут итераторами.
class SimpleIterator:
def __init__(self, limit):
self.limit = limit
self.counter = 0


def __next__(self):
if self.counter < self.limit:
self.counter += 1
return 1
else:
raise StopIteration

s_iter1 = SimpleIterator(3)

print(next(s_iter1))
print(next(s_iter1))
print(next(s_iter1))
print(next(s_iter1))
В нашем примере при четвертом вызове функции next() будет выброшено исключение StopIteration. Если мы хотим, чтобы с данным объектом можно было работать в цикле for, то в класс SimpleIterator нужно добавить метод __iter__(), который возвращает итератор, в данном случае этот метод должен возвращать self.
class SimpleIterator:
def __iter__(self):
return self


def __init__(self, limit):
self.limit = limit
self.counter = 0


def __next__(self):
if self.counter < self.limit:
self.counter += 1
return 1
else:
raise StopIteration

s_iter2 = SimpleIterator(5)

for i in s_iter2:
print(i)