python之flatten操作

例如:

复制代码li=[[1,2],[3,4],[5,6]] print [j for i in li for j in i] #or

1
2
3
li=[[1,2],[3,4],[5,6]]
print [j for i in li for j in i]
#or

复制代码from itertools import chain print list(chain(*li)) #or a=[[1,2],[3,4],[5,6]] t=[] [t.extend(i) for i in a] print t #or

1
2
3
4
5
6
7
8
from itertools import chain
print list(chain(*li))
#or
a=[[1,2],[3,4],[5,6]]
t=[]
[t.extend(i) for i in a]
print t
#or

复制代码print sum(li,[])

1
print sum(li,[])

对于复杂一些的，如：li=[1,[2],[[3]],[[4,[5],6]]]，上面的方法就不好使了，得换个方法了，
从结构上看像是树状的，很容易联想到了目录的遍历，于是就有了下面的做法：

复制代码

1