Python-提取最里面的列表

发布于 2021-01-29 16:02:05

刚开始玩弄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):

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

关注者
0
被浏览
37
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    假设没有像这样的“混合”列表,这似乎可行[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]
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看