Python嵌套列表转一维(压平嵌套列表)

       前一段去云英面试,技术官很”不厚道“了问了一个非常简单的问题:如何将多维列表转化了一维的?当时虽然想到了使用迭代或者列表生成式可以做到,但是可以没能可行的代码,回来后一顿后悔。

     对于规范的且嵌套维度较低的多维列表,python中有很多方法可以实现:

    a=[[1,2],[3,4],[5,6]]  
    print [j for i in li for j in i]  
    #or  
    from itertools import chain  
    print list(chain(*a))  
    #or  
    import itertools  
    a = [[1,2,3],[4,5,6], [7], [8,9]]  
    out = list(itertools.chain.from_iterable(a))  
    #or  
    a=[[1,2],[3,4],[5,6]]  
    t=[]  
    [t.extend(i) for i in a]  
    print t  
    #or  
    a=[[1,2],[3,4],[5,6]]  
    print sum(a,[])  
    #or  
    reduce(lambda x, y: x+ y, a)  

   对于复杂的多维嵌套列表就需要使用一些复杂的方法:

    #递归的方法比较容易理解  
    def expand_list(nested_list):  
        for item in nested_list:  
            if isinstance(item, (list, tuple)):  
                for sub_item in expand_list(item):  
                    yield sub_item  
            else:  
                yield item  
                  
    #在stackoverflow看到大牛的列表生成式版本  
    func = lambda x: [y for l in x for y in func(l)] if type(x) is list else [x]  
    #生成式的方法还有很多,可以自行摸索  
    对于使用python2.X的童鞋,还可以使用flatten函数来做:
    >>> d=[1,2,3,[4,5],[6,7,[8,9]]]  
    >>>from compiler.ast import flatten  
    >>>flatten(a)  
    [1, 2, 3, 4, 5, 6,7,8,9]  
   其实上边列举的方法仅仅是冰山一角,大海下的冰山就等这童鞋们自行去寻找挖掘了。


上文为转载文章,原文出处:https://blog.csdn.net/acmer_ding/article/details/73195260

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值