Sortir de plusieurs boucles for imbriquées en Python


Le mot clé break permet de sortir d’une boucle for abruptement. Mais une seule. Parfois on a 3, 4 boucles imbriquées, et on aimerait tellement sortir de toutes d’un coup.

Ce que je vais vous montrer est mal. Mais c’est tellement bon.

from contextlib import contextmanager
 
# on fait une exception qui hérite de StopIteration car c'est ce qui est utilisé
# de toute façon pour arrêter une boucle
class MultiStopIteration(StopIteration):
    # la classe est capable de se lever elle même comme exception
    def throw(self):
        raise self
 
 
@contextmanager
def multibreak():
    # le seul boulot de notre context manager c'est de donne le moyen de lever
    # l'exception tout en l'attrapant
    try:
        yield MultiStopIteration().throw
    except MultiStopIteration:
        pass

En gros on se créé un petit context manager, dont le seul but est de créer une exception qui va remonter en pêtant toutes les boucles. Je vous avais dit que c’était mal

Ca s’utilise comme ça:

>>> with multibreak() as stop:
...     for x in range(1, 4):
...         for z in range(1, 4):
...             for w in range(1, 4):
...                 print w
...                 if x * z * w == 2 * 2 * 2:
...                     print 'stop'
...                     stop() # appel MultiStopIteration().throw()
...
1
2
3
1
2
3
1
2
3
1
2
3
1
2
stop

Je vous avais dit que ça serait bon.

A part le fait que ce n’est pas très rapide au moment du bubbling de l’exception sur 3 blocks, il n’y a aucun danger ou side-effect. On triche en fait à peine, car le mécanisme interne des boucles en Python utilise de toute façon déjà une exception (StopIteration) pour dire à une boucle quand s’arrêter.

Bref, encore une victoire de connard.

10 thoughts on “Sortir de plusieurs boucles for imbriquées en Python

  • roro

    Trés bon ça. Je le range avec le pied de biche.
    Si t’en a d’autres, hésite pas.

    • Sam Post author

      @c0da. J’étais bourré quand j’ai écris l’article. Et puis c’est “avaiS dit” d’abord ! :-p

  • Christophe Simonis

    Sérieusement?

    La solution clean est de sortir toutes ces boucles dans une fonction et d’utiliser `return`.

  • Sam Post author

    Ou encore mieux d’utiliser un générateur et itertools.takewhile. Mais la sodomie n’est pas propre non plus, c’est pas pour ça que de temps en temps…

  • Kontre

    Je pense qu’il parle d’un des exemples donnés dans le pep 3136, où on ne lève qu’une seule exception:

    class BreakOutOfALoop(Exception): pass
     
    try:
        for a in a_list:
            for b in b_list:
                if condition_one(a, b):
                    raise BreakOutOfALoop
    except BreakOutOfALoop:
        break

    (Meh, ils sont où les espaces, et la coloration ?)

  • Kontre

    (Ah, les espaces et la coloration syntaxique ne se voient pas dans la prévisualisation)

  • buffalo974

    rajouter la ligne suivante en haut du script de Sam:

    from contextlib import contextmanager

Comments are closed.

Des questions Python sans rapport avec l'article ? Posez-les sur IndexError.