Question

刚开始使用Python，所以请耐心等待我：）

假设以下列表包含嵌套列表：

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

用不同的表示形式：

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

您将如何提取这些内部列表，以便返回包含以下表单的结果：

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

非常感谢！

编辑（感谢@falsetru）：

空内部列表或混合类型列表永远不会成为输入的一部分。

Answer 1

这似乎有效，假设没有像[1,2,[3]]这样的“混合”列表：

def get_inner(nested):
    if all(type(x) == list for x in nested):
        for x in nested:
            for y in get_inner(x):
                yield y
    else:
        yield nested

list(get_inner(nested_list))的输出：

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

甚至更短，没有生成器，使用sum来组合生成的列表：

def get_inner(nested):
    if all(type(x) == list for x in nested):
        return sum(map(get_inner, nested), [])
    return [nested]

Answer 2

使用itertools.chain.from_iterable：

from itertools import chain

def get_inner_lists(xs):
    if isinstance(xs[0], list): # OR all(isinstance(x, list) for x in xs)
        return chain.from_iterable(map(get_inner_lists, xs))
    return xs,

使用isinstance(xs[0], list)代替all(isinstance(x, list) for x in xs)，因为没有混合列表/空内部列表。

>>> list(get_inner_lists([[[[[1, 3, 4, 5]], [1, 3, 8]], [[1, 7, 8]]], [[[6, 7, 8]]], [9]]))
[[1, 3, 4, 5], [1, 3, 8], [1, 7, 8], [6, 7, 8], [9]]

Answer 3

比递归更有效：

result = []
while lst:
    l = lst.pop(0)
    if type(l[0]) == list:
        lst += [sublst for sublst in l if sublst] # skip empty lists []
    else:
        result.insert(0, l)

Python - 提取最内层列表

3 个答案: