Início / coding / Perguntas / 79448739
Accepted
Kiffikiffe
Asked: 2025-02-18 23:12:16 +0800 CST

Achatar uma lista multidimensional [duplicado]

  • 772

Digamos que temos uma lista multidimensional, mas com dimensões aleatórias, como:

[
    [
        [1, 2, [3, 4]],
        [[5, 6], 7]
    ],
    [8, 9, [10]]
]

Existe alguma maneira rápida de simplificar tudo e obter apenas a lista [1, 2, ..., 10]?

Sei que há uma solução para listas de listas, como loops ou listas de compreensão, mas aqui assumimos que não sabemos as dimensões da lista e que pode haver diferentes níveis de listas aninhadas.

python

2 respostas

  1. Best Answer

    Uma função recursiva pode fazer isso rapidamente:

    test = [
    [
        [1, 2, [3, 4]],
        [[5, 6], 7]
    ],
    [8, 9, [10]]
]

def flatten(l):
    for item in l:
        if isinstance(item, list):
            for i in flatten(item):
                yield i
        else:
            yield item

print(list(flatten(test)))

    [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

    Esta função itera por cada item na primeira camada de listas. Se encontrar algo que não seja uma lista, ela retorna com yield. Se encontrar outra lista, a função chama a si mesma para iterar sobre essa lista também. O que você acaba tendo é cada valor dentro de cada lista retornado de volta para o chamador.

    • 4

  2. Não importa quais são as dimensões. Simplesmente faça um achatamento recursivo que deve funcionar:

    def flatten(lst):
    flat_list = []
    for item in lst:
        if isinstance(item, list):
            flat_list.extend(flatten(item))
        else:
            flat_list.append(item)
    return flat_list
    • 2

relate perguntas