Что такое генератор и как его создать?

Практика задач для собеседования по программированию на Python

Kirill Smirnov

Data Science Consultant, Altran

Определение

Генератор — специальный итерируемый объект, создаваемый функцией с ключевым словом yield в теле.

def func():
    # Return a value from super complex calculations
    return 0

result = func()
print(result)

0
Практика задач для собеседования по программированию на Python

Определение

Генератор — специальный итерируемый объект, создаваемый функцией с ключевым словом yield в теле.

def func():
    # Yield a value from super complex calculations
    yield 0

result = func()
print(result)

<generator object result at 0x105736e10>
Практика задач для собеседования по программированию на Python

Генератор как итерируемый объект

def func():
    # Yield a value from super complex calculations
    yield 0

result = func()

for item in result:
    print(item)

0
Практика задач для собеседования по программированию на Python

Несколько yield!

def func():
    yield 0
    yield 1
    yield 2

result = func()
for item in result:
    print(item)

0
1
2
Практика задач для собеседования по программированию на Python

yield в цикле

def func(n):
    for i in range(0, n):
        yield 2*i

result = func(3)
for item in result:
    print(item)

0
2
4
Практика задач для собеседования по программированию на Python

Преобразование генератора в список

def func(n):
    for i in range(0, n):
        yield 2*i

result = func(5)

list(result)

[0, 2, 4, 6, 8]
Практика задач для собеседования по программированию на Python

Генератор как итератор

Генератор — это итерируемый объект И итератор

def func(n):
    for i in range(0, n):
        yield 2*i

result = func(3)

next(result)

0

next(result)

2

next(result)

4

next(result)

StopIteration
Практика задач для собеседования по программированию на Python

Генераторы одноразовые

def func(n):
    for i in range(0, n):
        yield 2*i

result = func(3)

for item in result:
    print(item)

0
2
4

for item in result:
    print(item)

# nothing

result = func(3)
for item in result:
    print(item)

0
2
4
Практика задач для собеседования по программированию на Python

Генераторы одноразовые

def func(n):
    for i in range(0, n):
        yield 2*i

result = func(3)
list(result)

[0, 2, 4]

list(result)

[]

result = func(3)
list(result)

[0, 2, 4]
Практика задач для собеседования по программированию на Python

Генераторное выражение

result = [2*i for i in range(0, 3)]
print(result)

[0, 2, 4]

result = (2*i for i in range(0, 3))
print(result)

<generator object result at 0x105736e10>
Практика задач для собеседования по программированию на Python

Обход элементов

result = (2*i for i in range(0, 3))

for item in result:
    print(item)

0
2
4

next(result)

StopIteration
Практика задач для собеседования по программированию на Python

Зачем нужны генераторы?

  • простой способ создать пользовательский итерируемый объект

[1, 3, 2, 4, 3, 5]

def create_jump_sequence(n):
    for i in range(1, n-1):
        yield i
        yield i+2

jump_sequence = create_jump_sequence(5)
list(jump_sequence)

[1, 3, 2, 4, 3, 5]
Практика задач для собеседования по программированию на Python

Зачем нужны генераторы?

  • простой способ создать пользовательский итерируемый объект
  • ленивая инициализация

[1, 3, 2, 4, 3, 5, 4, 6, 5, 7, ...]

def create_jump_sequence(n):
    for i in range(1, n-1):
        yield i
        yield i+2

jump_sequence = create_jump_sequence(500)
next(jump_sequence)

1
Практика задач для собеседования по программированию на Python

Зачем нужны генераторы?

  • простой способ создать пользовательский итерируемый объект
  • ленивая инициализация
  • возможность создавать бесконечные итерируемые объекты
def create_inf_generator():
    while True:
        yield 'I am infinite!'

inf_generator = create_inf_generator()

next(inf_generator)

I am infinite
Практика задач для собеседования по программированию на Python

Давайте потренируемся!

Практика задач для собеседования по программированию на Python

Preparing Video For Download...