Що таке генератор і як його створити?

Практика співбесід із програмування на 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

Генератор як ітератор

Генератор — це і Iterable, І Iterator

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)

# нічого

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...