Qu'est-ce qu'un générateur et comment en créer un ?
S'entraîner aux questions d'entrevue de programmation en Python
Kirill Smirnov
Data Science Consultant, Altran
Définition
Générateur — un objet itérable spécial créé par une fonction qui contient le mot-clé yield dans son corps.
def func():
# Return a value from super complex calculations
return 0
result = func()
print(result)
0
Définition
Générateur — un objet itérable spécial créé par une fonction qui contient le mot-clé yield dans son corps.
def func():
# Yield a value from super complex calculations
yield 0
result = func()
print(result)
<generator object result at 0x105736e10>
Générateur comme itérable
def func():
# Yield a value from super complex calculations
yield 0
result = func()
for item in result:
print(item)
0
Plusieurs yield
def func():
yield 0
yield 1
yield 2
result = func()
for item in result:
print(item)
0
1
2
Yield dans une boucle
def func(n):
for i in range(0, n):
yield 2*i
result = func(3)
for item in result:
print(item)
0
2
4
Convertir un générateur en liste
def func(n):
for i in range(0, n):
yield 2*i
result = func(5)
list(result)
[0, 2, 4, 6, 8]
Générateur comme itérateur
Un générateur est un Iterable ET un 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
Les générateurs sont consommables
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
Les générateurs sont consommables
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]
Compréhension de générateur
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>
Parcours
result = (2*i for i in range(0, 3))
for item in result:
print(item)
0
2
4
next(result)
StopIteration
Pourquoi des générateurs ?
- moyen simple de créer un objet itérable personnalisé
[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]
Pourquoi des générateurs ?
- moyen simple de créer un objet itérable personnalisé
- initialisation paresseuse
[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
Pourquoi des générateurs ?
- moyen simple de créer un objet itérable personnalisé
- initialisation paresseuse
- possibilité de créer des objets itérables infinis
def create_inf_generator():
while True:
yield 'I am infinite!'
inf_generator = create_inf_generator()
next(inf_generator)
I am infinite
Passons à la pratique !
S'entraîner aux questions d'entrevue de programmation en Python
