Programador Leigo
⚡ 60 segundos listas

Aplanar lista de listas em Python

Transforme uma lista de listas em uma lista simples com list comprehension ou itertools.chain.

flatten.py
matriz = [[1, 2], [3, 4], [5, 6]]

# Com list comprehension
plana = [x for sublista in matriz for x in sublista]
print(plana)  # [1, 2, 3, 4, 5, 6]

# Com itertools
from itertools import chain
plana2 = list(chain.from_iterable(matriz))
print(plana2)  # [1, 2, 3, 4, 5, 6]

Compartilhar