Ключевое слово yield в Python
Здравствуйте! В сегодняшней статье мы рассмотрим ключевое слово yield и создадим с его помощью функцию генератор. Но прежде скажем что из себя представляет генератор в Python. Генератор это процедура позволяющая управлять поведением цикла. Он сходен с функцией, но в отличие от нее не возвращает весь массив, а только одно значение за один раз, при каждом обращении к нему. Данная особенность позволяет экономить память.
И рассмотрим простую функцию:
def func():
return [1,2,3,4,5,6]
f = func()
print(f)
# вывод
[1, 2, 3, 4, 5, 6]
Данная функция возвращает весь список элементов. А чтобы вывести последовательно каждый элемент создадим функцию генератор:
def func_gen():
for i in [1,2,3,4,5]:
yield i
# присвоим результат функции новой переменной
f = func_gen()
print(next(f))
# вывод
1
print(next(f))
# вывод
2
Здесь используется функция итератора next(), которая позволяет выводить каждый последующий элемент за один раз. Выражение print(next(f)) будет выводить элементы из списка, до его окончания, при выходе за предел возникает исключение StopIteration.
Итерировать генераторы можно и в цикле for:
for i in func_gen():
print(i)
# вывод
1
2
3
4
5
6
Ключевое слово yield, используемое нами, возвращает значение и замораживает выполнение кода функции идущего после него. Каждое последующее обращение к функции будет запускать код с того места где он остановился. Т.е. при первом обращении к генератору функцией next() получим 1, при следующем вызове print(next(d)), 10 и 2 и т.д. так как именно в этом месте yield и заморозил нашу функцию:
def func_gen_two():
n = 10
for i in [1,2,3,4,5]:
yield i
print(n)
n +=10
d = func_gen_two()
print(next(d))
# вывод
1
print(next(d))
# вывод
10
2
for i in func_gen_two():
print(i)
# вывод
1
10
2
20
3
30
4
40
5
50
Таким образом мы выяснили что в Python можно создавать гибкие инструкции для управления итерациями цикла с помощью yield, и создали на его основе генератор, позволяющий экономить память при работе с итерируемыми данными.
-
- Михаил Русаков
Комментарии (0):
Для добавления комментариев надо войти в систему.
Если Вы ещё не зарегистрированы на сайте, то сначала зарегистрируйтесь.