python 多层嵌套列表 转成一维列表

如以下的嵌套列表:

[1,[2,3],[4,[5,6]]]

因为不知道嵌套了多少层,常规的方式是写一个递归函数,对这个函数进行递归来生成一维列表。

后来发现,在python的compiler模块中,提供了类似功能的函数:

>>> from compiler.ast import flatten
>>> flatten(a)
[1, 2, 3, 4, 5, 6]

这样子就简单多了

猜你喜欢

转载自blog.csdn.net/u011085172/article/details/81165698