博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
PYTHON压平嵌套列表
阅读量:7261 次
发布时间:2019-06-29

本文共 1341 字,大约阅读时间需要 4 分钟。

list 是 Python 中使用最频繁的数据类型, 标准库里面有丰富的函数可以使用。

不过,如果把多维列表转换成一维列表(不知道这种需求多不多),还真不容易找到好用的函数,
要知道Ruby、Mathematica、Groovy中可是有flatten的啊。
如果列表是维度少的、规则的,还算好办
例如:

li=[[1,2],[3,4],[5,6]]print [j for i in li for j in i]#orfrom itertools import chainprint list(chain(*li))#ora=[[1,2],[3,4],[5,6]]t=[][t.extend(i) for i in a]print t#orprint 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 _flattenli=reduce(lambda *x:list(x),range(2,6),[1])print liprint _flatten(li)#Out:#[[[[[1], 2], 3], 4], 5]#(1, 2, 3, 4, 5)#对元组同样适用

 

还有一些第三方模块提供这样的功能,如sympy、numpy、pipe等

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

FROM: http://www.cnblogs.com/c-hy/archive/2012/09/21/2696703.html

转载于:https://www.cnblogs.com/Vito2008/p/5101787.html

你可能感兴趣的文章
「算法」如何实现大整数相乘?(下)
查看>>
Oracle总结【SQL细节、多表查询、分组查询、分页】
查看>>
具有代表性的 HTTP 状态码
查看>>
iOS 组件化 —— 路由设计思路分析
查看>>
扯扯ID
查看>>
mp-redux:解耦小程序中的业务与视图,让测试更容易
查看>>
Sql注入
查看>>
如何用Python写一个贪吃蛇AI
查看>>
Docker 镜像优化与最佳实践
查看>>
谁动了我的 DOM??!
查看>>
web圣杯布局
查看>>
玩转Koa -- koa-router原理解析
查看>>
数据结构和算法面试题系列—数字题总结
查看>>
通过一个场景实例 了解前端处理大数据的无限可能
查看>>
快速体验k8s集群的测试、开发环境--allinone部署,国内环境下觉得比官方的minikube方便、简单很多。...
查看>>
填坑之Hexo博客搭建
查看>>
Elasticsearch学习记录第二篇
查看>>
android 解码混淆过的堆栈信息
查看>>
[翻译]Scope in JavaScript
查看>>
Android逆向(一) —— AndroidManifest.xml 二进制解析
查看>>