python - 递归地展平列表

Question

可能重复：
在 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

好吧，如果你想要一种 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)]))

Answer 4

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

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]