Как создать генератор

Генератор в Python — это функция, которая использует ключевое слово yield вместо return для возврата итератора. При вызове генератор возвращает объект-генератор, который можно итерировать.

Пример:

def countdown(n):
    while n > 0:
        yield n
        n -= 1

# Использование
for num in countdown(5):
    print(num)  # 5, 4, 3, 2, 1

Особенности:

• Генератор сохраняет состояние между вызовами.
• Выполняется лениво (по требованию).
• Потребляет меньше памяти, чем списки.

Можно создать генераторное выражение:

squares = (x**2 for x in range(10))