Comment puis-je aplatir les résultats de compréhension de listes imbriquées en Python ?

Aplatissement des résultats de compréhension de liste

Lors de l'utilisation de compréhensions de liste, il est courant de rencontrer des listes imbriquées en sortie, en particulier lorsque vous travaillez avec des fonctions qui renvoient des listes . Pour convertir ces listes imbriquées en un résultat plat, envisagez d'implémenter les techniques suivantes :

Itérations imbriquées dans les compréhensions de liste

Les compréhensions de liste de Python prennent en charge les boucles imbriquées. Cela vous permet de parcourir plusieurs collections et d'aplatir les résultats. Par exemple, la compréhension de liste imbriquée suivante convertit une liste de répertoires en une liste plate de sous-répertoires :

<code class="python">dirs = ['dir1', 'dir2', 'dir3']
filenames = [filename for path in dirs for filename in os.listdir(path)]</code>

Cette compréhension parcourt les répertoires, accédant à chaque chemin de répertoire. Dans la boucle interne, il utilise os.listdir pour récupérer les noms de fichiers dans ce répertoire. La liste résultante contient tous les sous-répertoires.

Chaînage des compréhensions de liste

Vous pouvez également enchaîner plusieurs compréhensions de liste pour obtenir un aplatissement. Cette approche convient lorsque vous avez plusieurs transformations à appliquer :

<code class="python">directories = ['dir1', 'dir2', 'dir3']
filenames = [filename for path in directories if os.path.isdir(path) for filename in os.listdir(path)]</code>

Dans cet exemple, nous filtrons d'abord les non-répertoires des répertoires (en utilisant la clause if). Ensuite, nous appliquons la même opération os.listdir que dans l’exemple précédent. L'enchaînement de ces compréhensions aplatit les résultats.

En employant des itérations imbriquées ou en chaînant des compréhensions, vous pouvez aplatir efficacement les résultats de compréhension de listes imbriquées, offrant ainsi un moyen pratique de simplifier votre code et de maintenir la lisibilité.

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!