Question

可能重复：
Flatten (an irregular) list of lists in Python

我无法使用python递归展平列表。我已经看到了许多方法需要列表理解和需要导入的各种方法但是我正在寻找一种非常基本的方法来递归地展平不使用任何for循环的不同深度的列表。我进行了一系列测试，但有两个我无法通过

flatten([[[[]]], [], [[]], [[], []]]) # empty multidimensional list
flatten([[1], [2, 3], [4, [5, [6, [7, [8]]]]]]) # multiple nested list

我的代码

def flatten(test_list):
    #define base case to exit recursive method
    if len(test_list) == 0:
       return []
    elif isinstance(test_list,list) and type(test_list[0]) in [int,str]:
        return [test_list[0]] + flatten(test_list[1:])
    elif isinstance(test_list,list) and isinstance(test_list[0],list):
        return test_list[0] + flatten(test_list[1:])
    else:
        return flatten(test_list[1:])

我很感激一些建议。

Answer 1

这可以处理你的两种情况，我认为将解决一般情况，没有任何for循环：

def flatten(S):
    if S == []:
        return S
    if isinstance(S[0], list):
        return flatten(S[0]) + flatten(S[1:])
    return S[:1] + flatten(S[1:])

Answer 2

li=[[1,[[2]],[[[3]]]],[['4'],{5:5}]]
flatten=lambda l: sum(map(flatten,l),[]) if isinstance(l,list) else [l]
print flatten(li)

Answer 3

这是一个可能的解决方案，没有任何循环或列表推导，只使用递归：

def flatten(test_list):
    if isinstance(test_list, list):
        if len(test_list) == 0:
            return []
        first, rest = test_list[0], test_list[1:]
        return flatten(first) + flatten(rest)
    else:
        return [test_list]

Answer 4

好吧，如果你想要一个lisp方式，让我们拥有它。

atom = lambda x: not isinstance(x, list)
nil  = lambda x: not x
car  = lambda x: x[0]
cdr  = lambda x: x[1:]
cons = lambda x, y: x + y

flatten = lambda x: [x] if atom(x) else x if nil(x) else cons(*map(flatten, [car(x), cdr(x)]))

递归地展平列表

4 个答案: