好得很程序员自学网

<tfoot draggable='sEl'></tfoot>

PYTHON压平嵌套列表

PYTHON压平嵌套列表

list 是 Python 中使用最频繁的数据类型, 标准库里面有丰富的函数可以使用。
不过,如果把多维列表转换成一维列表(不知道这种需求多不多),还真不容易找到好用的函数,
要知道Ruby、Mathematica、Groovy中可是有flatten的啊。
如果列表是维度少的、规则的,还算好办
例如:

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()]

不过,这种做法对于列表中出现包含"["或"]"的字符串时就无能为力了,需要改进.


其他方法:

国外某论坛上见到的,同样是递归,一行搞定

flat= lambda  L: sum(map(flat,L),[])  if  isinstance(L,list)  else   [L 

下面这个方法用到Tkinter模块,在邮件列表看到的方法。估计很多同学还不知道它能办到吧,也算是python自带。注意,windows版的python都自带Tkinter模块的,linux默认则没有

 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等

对于嵌套的元组,无需多说了吧,只需稍加改动就可以了

 

 

标签:  python list

作者: Leo_wl

    

出处: http://www.cnblogs.com/Leo_wl/

    

本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。

版权信息

查看更多关于PYTHON压平嵌套列表的详细内容...

  阅读:40次