一. 先来一个简单的实现
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
print sum(li,[])
但这只能对简单的二维列表进行压平,对于复杂一些的,如:li=[1,[2],[[3]],[[4,[5],6]]],上面的方法就不好使了,得换个方法了,从结构上看像是树状的,很容易联想到了目录的遍历,于是就有了下面的做法:
思路二:目录遍历
def flat(tree):
res = []
for i in tree:
if isinstance(i, list):
res.extend(flat(i))
else:
res.append(i)
return res
思路三:转变为字符串,去掉中间的括号
另一种思路,嵌套列表无非就是有很多成对的方括号,一维的列表只有一对,把中间的去掉就行了,转换为字符串就好办了。
def flatten(seq):
s=str(seq).replace('[', '').replace(']', '') #当然也可以用正则
return [eval(x) for x in s.split(',') if x.strip()]
不过,这种做法对于列表中出现包含"["或"]"的字符串时就无能为力了,需要改进.
思路四:Lambda一行代码
flat=lambda L: sum(map(flat,L),[]) if isinstance(L,list) else [L]
思路五:TKinter模块
from Tkinter import _flatten
li=reduce(lambda *x:list(x),range(2,6),[1])
print li
print _flatten(li)
#Out:
#[[[[[1], 2], 3], 4], 5]
#(1, 2, 3, 4, 5)
#对元组同样适用
其他
还有一些第三方模块提供这样的功能,如sympy、numpy、pipe等对于嵌套的元组,无需多说了吧,只需稍加改动就可以了。比如numpy.ndarray有flatten。Sympy中也有flatten()函数。
网友评论