python压平嵌套列表

枫铃3年前 (2021-07-23)Python260

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]]],上面的方法就不好使了,得换个方法了,
从结构上看像是树状的,很容易联想到了目录的遍历,于是就有了下面的做法:

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:778463939
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
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()]

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

相关文章

一篇文章带你搞定Python返回函数

一篇文章带你搞定Python返回函数

一、什么是返回函数?...

超详细 1小时学会Python输入和输出!

超详细 1小时学会Python输入和输出!

1.Hello world 安装完Python之后,打开IDLE(Python GUI) ,该程序是Python语言解释器,你写...

python 将文件描述符包装成文件对象

有一个对应于操作系统上一个已打开的I/O 通道(比如文件、管道、套接字等)的整型文件描述符,你想将它包装成一个更高层的Python 文件对象。...

python字典的增,删,改,查

字典—dict 1.字典是无序,可变的数据类型 2.字典:用于存储数据,存储大量数据,字典要比列表快,...

发表评论

访客

看不清,换一张

◎欢迎参与讨论,请在这里发表您的看法和观点。