Comment pouvons-nous aplatir efficacement des listes arbitrairement imbriquées en Python ?

Aplatissement arbitraire des listes imbriquées

Les listes imbriquées, avec différents niveaux de profondeur, présentent un défi en les réduisant à une seule dimension. Bien qu'il existe de nombreuses solutions pour aplatir les listes superficielles, beaucoup ont du mal avec les listes irrégulièrement imbriquées, telles que [[[1, 2, 3], [4, 5]], 6].

Le système récursif Approche

Une approche, comme mentionné dans la question, implique la récursion :

def flatten(x):
    result = []
    for el in x:
        if hasattr(el, "__iter__") and not isinstance(el, basestring):
            result.extend(flatten(el))
        else:
            result.append(el)
    return result

Cet algorithme itère à travers la liste, en aplatissant de manière récursive toutes les listes imbriquées et en ajoutant des éléments non itérables au résultat.

Fonctions de générateur pour une lisibilité et des performances améliorées

Les fonctions de générateur offrent une approche alternative qui peut améliorer à la fois la lisibilité et l'efficacité de notre processus d'aplatissement.

Python 2

from collections import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, basestring):
            for item in flatten(x):
                yield item
        else:
            yield x

Python 3

from collections.abc import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, (str, bytes)):
            yield from flatten(x)
        else:
            yield x

Dans Python 3, l'opérateur rendement de renvoie commodément les éléments des générateurs imbriqués de manière séquentielle, tandis que dans Python 2 , nous parcourons explicitement les sous-générateurs.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!