Python - "Unlist"列表中的嵌套元素

Question

这个问题已经有答案了:

Flatten an irregular (arbitrarily nested) list of lists (51 个回答)

已关闭 5 年前。

有没有快速的方法来理清列表中的元素？例如:鉴于

list = [[1,2,3],[4,2],2,1,3]

我们将有:

list = [1,2,3,4,2,2,1,3]

Answer 1

将任何非列表值转换为列表后，您可以使用列表理解:

l = [[1,2,3],[4,2],2,1,3]
new_l = [i for b in map(lambda x:[x] if not isinstance(x, list) else x, l) for i in b]

输出:

[1, 2, 3, 4, 2, 2, 1, 3]

编辑:对于嵌套级别，您可以使用带有生成器表达式的递归:

def flatten(d):
 v = [[i] if not isinstance(i, list) else flatten(i) for i in d]
 return [i for b in v for i in b]

打印(展平(l)输出:

[1, 2, 3, 4, 2, 2, 1, 3]