美文网首页
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层嵌套的列表

    网络上公布了很多列表嵌套展开的语法,这里就不一一列举了,方法大致包含 lambda语句, itertools. 这...

  • Checkin笔记 - Flatten a List

    题目 将一个嵌套的列表展开成一维列表,嵌套的层数不固定。例如,输入一个 列表[[[2]], [4, [5, 6, ...

  • markdown test

    一级标题 二级标题 三级标题 嵌套列表嵌套列表嵌套列表嵌套列表嵌套列表嵌套列表 无序列表 无序列表 有序列表1 有...

  • HTML学习之列表

    通过HTML实现如下列表 可见该列表为嵌套列表,最外一层为无序列表,第二层为有序列表,第三层为无序列表,并且无序列...

  • 【慕课网】Python入门笔记 「4」基本数据类型

    【慕课网】Python入门笔记 「4」“组”的概念与定义 [TOC] 1.列表的定义(数组) 列表: 嵌套列表(其...

  • Python学习(二)--列表、元组、集合

    1.列表的定义 列表里可以有多种类型,列表里也可以有列表,在其他语言里叫二维数组,对于Python可以叫做嵌套列表...

  • 一 -18 python (基础)列表的镶嵌

    列表嵌套 类似while循环的嵌套,列表也是支持嵌套的 一个列表中的元素又是一个列表,那么这就是列表的嵌套叫做二维...

  • Python基础_04:列表(2019-1-13)

    列表 列表嵌套

  • python--深浅拷贝

    python学习中,有这样一个问题,列表的拷贝。当列表有嵌套列表时候,那么在进行复制拷贝的时候就会产生内列表的元素...

  • python--嵌套序列

    python 嵌套序列 嵌套序列:是由一组等长列表组成的列表。 numpy可以对数据进行类型转换, 统计转成字符型,,

网友评论

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

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