去除list列表中的多余维度(list squeeze)、嵌套展开

先说结论,

list不支持squeeze操作。

对于numpy中的array对象,则可以直接使用squeeze函数进行单维度删除

具体使用方法可参考这篇文章Numpy库学习—squeeze()函数_o_Eagle_o的博客-CSDN博客

不过很遗憾,list就没有这么丝滑的方法了。

但是对于嵌套的list,比如下面这种形式:

list = [[1, 2, 3], [4, 5], [6]] 

如果想要将其转为:

[1,2,3,4,5,6]

则可以使用如下三种方式:

# 方法一:
list_1 = [[1, 2, 3], [4, 5], [6]]
list_2 = []
for element in list_1:
    list_2 += element

 
# 方法二:列表推导
list_1 = [[1, 2, 3], [4, 5], [6]]
list_2 = [value for sub_list in list_1 for value in sub_list]

 
# 方法三:使用sum
list_1 = [[1, 2, 3], [4, 5], [6]]
list_2 = sum(list_1, [])

不过,对于比较复杂的形式,比如

list = [1, [2], [[3]], [[4, [5], 6]], 7, [8]]

上述三种方法就无法解决了,可以通过递归的方式实现,

定义函数如下:

def flat(lists):
    res = []
    for candidate in lists:
        if isinstance(candidate, list):
            res.extend(flat(candidate))
        else:
            res.append(candidate)
    return res

参考:

Python 嵌套列表展开_51CTO博客_python列表嵌套列表

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值