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