La fonction yield

Le mot-clé yield et les générateurs

En Python, la plupart des fonctions calculent un résultat et le retournent en une seule fois avec return. Mais que faire lorsqu'on veut produire une séquence de valeurs sans les stocker toutes en mémoire ? C'est le rôle de yield.

def generateur():
    # ...
    yield valeur      # suspend et renvoie valeur
    # ... reprend ici à la prochaine itération

return vs yield

Exemples

Exemple — Générateur simple de 0 à n-1

def generateur(n):
    for i in range(n):
        yield i         # suspend et renvoie i à chaque itération

for valeur in generateur(6):
    print(valeur)

0
1
2
3
4
5

Voici ce qui se passe à chaque itération de la boucle for :

1. La boucle appelle next() sur le générateur.
2. Le générateur s'exécute jusqu'au prochain yield.
3. Il renvoie la valeur et se met en pause.
4. À l'itération suivante, il reprend exactement après le yield.
5. Quand la fonction se termine, une exception StopIteration est levée automatiquement.

Exemple — Générateur de factorielles (2 → n)

def factorielles(n):
    f = 1
    for i in range(2, n):
        f *= i
        yield f         # renvoie la factorielle courante

for val in factorielles(6):
    print(val)

2
6
24
120

gen = factorielles(6)
print(next(gen))   # 2
print(next(gen))   # 6
print(next(gen))   # 24

Exemple — Générateur infini

Contrairement aux listes, un générateur peut produire une séquence théoriquement infinie, car il ne calcule qu'une valeur à la fois.

def entiers_naturels():
    n = 0
    while True:         # boucle infinie, mais sans risque mémoire
        yield n
        n += 1

gen = entiers_naturels()
for _ in range(5):
    print(next(gen))    # 0, 1, 2, 3, 4

Avantages et inconvénients

Applications pratiques

Les générateurs sont particulièrement adaptés dans les contextes suivants :

Exemple — Lire un grand fichier ligne par ligne

def lire_lignes(chemin):
    with open(chemin, 'r', encoding='utf-8') as f:
        for ligne in f:
            yield ligne.strip()   # une ligne à la fois, sans tout charger

for ligne in lire_lignes("donnees.txt"):
    print(ligne)

Exemple — Suite de Fibonacci infinie

def fibonacci():
    a, b = 0, 1
    while True:
        yield a
        a, b = b, a + b

gen = fibonacci()
for _ in range(8):
    print(next(gen), end=" ")   # 0 1 1 2 3 5 8 13

0 1 1 2 3 5 8 13