python 列表展开,适用于N层嵌套的列表

我的简书地址:http://www.jianshu.com/p/8215f455e4e5
网络上公布了很多列表嵌套展开的语法,这里就不一一列举了,方法大致包含 lambda语句, itertools. 这些工具存在问题,就是不能对N(N>3)的列表进行展开。曾经最喜欢用的神器 from compiler.ast import flatten 可以解决这个问题,但是现在python3 不支持这个包了。 试来试去,索性不如自己写一个。

def flatten(input_list):
    output_list = []
    while True:
        if input_list == []:
            break
        for index, i in enumerate(input_list):

            if type(i)== list:
                input_list = i + input_list[index+1:]
                break
            else:
                output_list.append(i)
                input_list.pop(index)
                break

    return output_list

代码的原理比较简单,伪代码是:
1. 遍历原始列表
2. 对于列表中的当前元素
2.1 如果它是非列表元素,存入输出列表。
2.2 如果它是列表元素:进入该列表,执行2.1

一句话说清楚就是“遍历原始列表,对于列表内的元素,如果非列表就存入输出列表中,如果是列表元素,则进入改列表元素并查看列表内元素类型”

测试一下:

a = [ 1, 2, [3, 4, [5, 6, [7, 8, [9, [10, ['end'] ] ] ] ] ] ]

demo for the flatten() function

完全没毛病,支持各种数据类型。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值