python 列表展开，适用于N层嵌套的列表

我的简书地址：http://www.jianshu.com/p/8215f455e4e5

网络上公布了很多列表嵌套展开的语法，这里就不一一列举了，方法大致包含 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

代码的原理比较简单，伪代码是：

1. 遍历原始列表

2. 对于列表中的当前元素

2.1 如果它是非列表元素，存入输出列表。

2.2 如果它是列表元素：进入该列表，执行2.1

一句话说清楚就是“遍历原始列表，对于列表内的元素，如果非列表就存入输出列表中，如果是列表元素，则进入改列表元素并查看列表内元素类型”

测试一下：

a = [ 1, 2, [3, 4, [5, 6, [7, 8, [9, [10, ['end'] ] ] ] ] ] ]

完全没毛病，支持各种数据类型。