Python - 提取最内层列表

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）：

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

- Roi Tal

---

[[[[[1, 3, 4, 5]], [1, 3, 8]], [[1, 7, 8]]], [[[6, 7, 8]]], [9, [10]]]和[[[[[1, 3, 4, 5]], [1, 3, 8]], [[1, 7, 8]]], [[[6, 7, 8]]], []]应该返回什么？ - falsetru

感谢您的澄清问题： 一个空列表或者一个包含混合类型项的列表永远不会成为输入的一部分。 - Roi Tal

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]

- tobias_k

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]]

- falsetru

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) 

- user278064

---

从列表中删除第一个元素，将元素插入到列表开头需要 O(n) 的时间。使用 collections.deque 可以提高速度。请参见 http://ideone.com/RFGhnh。 - falsetru

如果你声称你的解决方案比其他解决方案更高效，请附上小型和大型输入的基准测试。请参考falsetru的评论，了解为什么它很慢。 - pts