Question

I want to know in python how to take a multidimensional array and put it in a 1D list. This thing work:

a = [[1,2],[3,4]]
sum(a,[])

I get this : [1,2,3,4]

But if I have a multidimensional with unknow dimension or not constant dimension like that :

a = [1,[3,4,[5,6,7]]]

How to get this : [1,2,3,4,5,6,7]

thanks

Answer 1

One way is to use recursion, something like this should work:

def flatten_list(l):
    new_l = []
    for item in l:
        if type(item) == type([]):
            new_l += flatten_list(item)
        else:
            new_l.append(item)
    return new_l

I did not test this code but the idea is there.