网络上公布了很多列表嵌套展开的语法,这里就不一一列举了,方法大致包含 lambda语句, itertools. 这些工具存在问题,就是不能对N(N>3)的列表进行展开。曾经最喜欢用的神器 from compiler.ast import flatten 可以解决这个问题,但是现在python3 不支持这个包了。 试来试去,索性不如自己写一个。
def flatten(input_list):
output_list = []
while True:
if input_list == []:
break
for index, i in enumerate(input_list):
if type(i)== list:
input_list = i + input_list[index+1:]
break
else:
output_list.append(i)
input_list.pop(index)
break
return output_list
代码的原理比较简单,伪代码是:
- 遍历原始列表
- 对于列表中的当前元素
2.1 如果它是非列表元素,存入输出列表。
2.2 如果它是列表元素:进入该列表,执行2.1
一句话说清楚就是“遍历原始列表,对于列表内的元素,如果非列表就存入输出列表中,如果是列表元素,则进入改列表元素并查看列表内元素类型"
测试一下:
a = [ 1, 2, [3, 4, [5, 6, [7, 8, [9, [10, ['end'] ] ] ] ] ] ]
demo for the flatten() function
完全没毛病,支持各种数据类型。
网友评论