美文网首页
python 列表展开,适用于N层嵌套的列表

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

作者: vola_lei | 来源:发表于2017-04-03 16:08 被阅读0次

    网络上公布了很多列表嵌套展开的语法,这里就不一一列举了,方法大致包含 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

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

    相关文章

      网友评论

          本文标题:python 列表展开,适用于N层嵌套的列表

          本文链接:https://www.haomeiwen.com/subject/nscoottx.html