Генераторы

Генераторы — это особый тип итераторов в Python, позволяющий лениво перебирать данные. Вместо возврата всех данных сразу они выдают по одному элементу за раз, эффективно используя память.

В этом уроке Вы узнаете, как создавать и использовать генераторы в Python.

Генератор — это функция, которая использует ключевое слово yield, чтобы возвращать значения по одному. Каждый вызов метода next() генератора возобновляет выполнение с места, где в последний раз было возвращено значение.

# Basic generator example
def my_generator():
    yield 1
    yield 2
    yield 3

# Using the generator
gen = my_generator()
print(next(gen))  # Outputs: 1
print(next(gen))  # Outputs: 2
print(next(gen))  # Outputs: 3

Генераторы сохраняют состояние функции, поэтому могут возобновить выполнение ровно с места остановки. Благодаря этому они эффективно используют память и особенно подходят для больших наборов данных.

# Example: Generating a range of numbers
def range_generator(start, end):
    while start < end:
        yield start
        start += 1

for number in range_generator(1, 5):
    print(number)  # Outputs: 1, 2, 3, 4

Выражения-генераторы похожи на включения списков, но вместо квадратных скобок используют круглые. Они создают генераторы, а не списки.

# Generator expression
gen_exp = (x * x for x in range(5))

for value in gen_exp:
    print(value)  # Outputs: 0, 1, 4, 9, 16

Генераторы эффективнее списков с точки зрения использования памяти, поскольку не хранят все данные в памяти. Вместо этого они создают данные по мере необходимости.

# Comparing lists and generators
# List comprehension
list_comp = [x * x for x in range(1000000)]

# Generator expression
gen_exp = (x * x for x in range(1000000))

# List comprehension uses more memory
print(type(list_comp))  # Outputs: <class 'list'>
print(type(gen_exp))    # Outputs: <class 'generator'>

Генераторы идеально подходят для:

• Обработки больших наборов данных, которые не помещаются в памяти.
• Создания бесконечных последовательностей.
• Отложенного вычисления данных.

Генераторы могут создавать бесконечные последовательности, например бесконечную последовательность чисел. Не забудьте остановить цикл, чтобы предотвратить бесконечное выполнение.

# Infinite generator example
def infinite_sequence():
    num = 0
    while True:
        yield num
        num += 1

# Using the infinite generator
for i in infinite_sequence():
    if i > 5:
        break
    print(i)  # Outputs: 0, 1, 2, 3, 4, 5

Функции-генераторы могут содержать несколько операторов yield, что позволяет создавать сложные последовательности:

# Multiple yields in a generator
def multi_yield():
    yield "First"
    yield "Second"
    yield "Third"

for value in multi_yield():
    print(value)  # Outputs: First, Second, Third

Вот несколько ошибок, которых следует избегать:

• Использование return вместо yield в функции-генераторе.
• Непотребление генератора с помощью цикла или next().
• Использование генераторов для небольших наборов данных, когда больше подходят списки.

В этом уроке Вы узнали:

• Что такое генераторы и как они работают в Python.
• Как использовать yield для создания функций-генераторов.
• О различиях между генераторами и списками.
• О вариантах использования генераторов, например для создания бесконечных последовательностей и работы с большими наборами данных.

Вы отлично справились! Перейдём к следующей теме.