Comment puis-je extraire le premier élément de chaque sous-liste dans une liste de listes Python ?

Extraire le premier élément de chaque sous-liste en Python

En Python, lorsque vous travaillez avec des structures de données imbriquées comme des listes de listes, cela peut être nécessaire pour extraire des éléments spécifiques de chaque sous-liste pour créer une nouvelle liste. Par exemple, pour extraire le premier élément de chaque sous-liste d'une liste de listes, vous pouvez utiliser les méthodes suivantes :

1. Compréhension de liste :

Cette approche concise utilise la compréhension de liste pour parcourir chaque sous-liste et ajouter le premier élément à une nouvelle liste.

lst = [['a', 'b', 'c'], [1, 2, 3], ['x', 'y', 'z']]
lst2 = [item[0] for item in lst]

2. Map et Lambda :

Vous pouvez utiliser la fonction map() avec une fonction lambda pour obtenir le même résultat. La fonction lambda récupère le premier élément de chaque sous-liste.

lst = [['a', 'b', 'c'], [1, 2, 3], ['x', 'y', 'z']]
lst2 = list(map(lambda sublist: sublist[0], lst))

3. Boucle For :

Bien que légèrement moins efficace que les méthodes précédentes, une boucle for peut également accomplir la tâche.

lst = [['a', 'b', 'c'], [1, 2, 3], ['x', 'y', 'z']]
lst2 = []
for sublist in lst:
    lst2.append(sublist[0])

Le choix approprié dépend des exigences spécifiques et des considérations de performances. de votre code. Pour des scénarios simples et efficaces, la compréhension de liste ou map() avec lambda sont généralement préférés.

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!