O que é um generator e como criar um?
Praticando questões de entrevista de código em Python
Kirill Smirnov
Data Science Consultant, Altran
Definição
Generator — um objeto iterável especial criado por uma função que tem a palavra-chave yield no corpo.
def func():
# Return a value from super complex calculations
return 0
result = func()
print(result)
0
Definição
Generator — um objeto iterável especial criado por uma função que tem a palavra-chave yield no corpo.
def func():
# Yield a value from super complex calculations
yield 0
result = func()
print(result)
<generator object result at 0x105736e10>
Generator como iterável
def func():
# Yield a value from super complex calculations
yield 0
result = func()
for item in result:
print(item)
0
Mais yields!
def func():
yield 0
yield 1
yield 2
result = func()
for item in result:
print(item)
0
1
2
Yield em loop
def func(n):
for i in range(0, n):
yield 2*i
result = func(3)
for item in result:
print(item)
0
2
4
Converter um generator em lista
def func(n):
for i in range(0, n):
yield 2*i
result = func(5)
list(result)
[0, 2, 4, 6, 8]
Generator como iterador
Generator é Iterável E Iterador
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
Generators são descartáveis
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
Generators são descartáveis
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]
Generator comprehension
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>
Percorrendo
result = (2*i for i in range(0, 3))
for item in result:
print(item)
0
2
4
next(result)
StopIteration
Por que generators?
- forma simples de criar um objeto iterável personalizado
[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]
Por que generators?
- forma simples de criar um objeto iterável personalizado
- inicialização preguiçosa
[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
Por que generators?
- forma simples de criar um objeto iterável personalizado
- inicialização preguiçosa
- possibilidade de criar objetos iteráveis infinitos
def create_inf_generator():
while True:
yield 'I am infinite!'
inf_generator = create_inf_generator()
next(inf_generator)
I am infinite
Vamos praticar!
Praticando questões de entrevista de código em Python
