Python中的部分列表展平

Question

我有一个小问题，我缺乏相当多的 Python 经验。 假设一个列表：

list=[[40, 20, 45, 40], [[20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]]

我想做一些我称之为“部分展平”的事情，因为预期的输出是：

[[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]

注意list 之前的长度为2，而预期输出的长度为3。问题是我事先不知道list 中的哪个元素嵌套更深一层。

This answer 和许多其他人并没有太大帮助，因为扁平化太远了。问题之后的输出是：

list=[40, 20, 45, 40, [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]

（请注意列表中前 4 个元素缺少括号。

Answer 1

您可以创建一个递归函数，测试all 内部元素是否为列表。如果是，则递归应用该算法，否则生成列表本身。

def inner_lists(lst):
    if all(isinstance(x, list) for x in lst):
        for inner in lst:
            for x in inner_lists(inner):
                yield x
    else:
        yield lst

>>> lst = [[40, 20, 45, 40], [[20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]]
>>> list(inner_lists(lst))
[[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]

或使用yield from（Python 3.3 及更高版本）：

def inner_lists(lst):
    if all(isinstance(x, list) for x in lst):
        yield from (x for inner in lst for x in inner_lists(inner))
    else:
        yield lst

或者没有生成器：

def inner_lists(lst):
    if all(isinstance(x, list) for x in lst):
        return [x for inner in lst for x in inner_lists(inner)]
    else:
        return [lst]

Answer 2

如果您的列表只能包含数字或子列表（不能同时包含两者），您可以使用以下生成器函数：

In [25]: def partial_flattening(lst):
             for sub in lst:
                 if isinstance(sub[0], list):
                     yield from sub # in python 2.x use for i in sub: yeild i
                 else:          
                     yield sub
   ....:                 

In [26]: list(partial_flattening(lst))
Out[36]: [[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]

如果可能有两种类型的组合，那么根据您想要对数字做什么，您可以自定义功能。

或者作为更简洁的方法，您可以在列表理解中使用yield：

In [29]: list([(yield from i) if isinstance(i[0], list) else (yield i) for i in lst])
Out[29]: [[40, 20, 45, 40], [20, 10, 10, 30, 45, 20], [30, 20, 20, 30]]