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

2017-04-03  本文已影响0人  vola_lei

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

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

上一篇 下一篇

猜你喜欢

热点阅读