python 中将一个任意嵌套的数字列表展开为一个列表

最新推荐文章于 2024-07-31 21:00:00 发布

张小李的风

最新推荐文章于 2024-07-31 21:00:00 发布

阅读量433

点赞数

文章标签： python 列表数据结构算法

本文链接：https://blog.csdn.net/qq_38941890/article/details/115626230

版权

这篇博客介绍了如何在Python中将一个多层嵌套的数字列表展开为一个平铺的列表。通过定义一个名为`flatten`的生成器函数，递归地处理列表中的每个元素，遇到子列表时继续展开，遇到非列表元素则直接yield。示例代码展示了如何使用这个函数处理两个嵌套列表`list1`和`list2`，并输出展开后的结果。

摘要由CSDN通过智能技术生成

@python 中将一个任意嵌套的数字列表展开为一个列表

问题

当导入的数据是多层嵌套的列表时，如何将其展开为一层的列表？

输入数据

list1 = [[1.3, 2, 3], [5.0, 6], [4], [9, 5, 4, 3]]
list2 = [[1], [1, [3], [5, [6]]]]

大量的括号会影响后续的操作，如何将其展开？
使用yield语句，构造一个生成器

代码

def flatten(listIn):
    try:
        for sublist in listIn:
            for element in flatten(sublist):
                yield element
    except TypeError:
        yield listIn

输出结构

[1.3, 2, 3, 5.0, 6, 4, 9, 5, 4, 3]
[1, 1, 3, 5, 6]

实例代码

# 将一个任意嵌套的数字列表展开
def flatten(nested):
    try:
        for sublist in nested:
            for element in flatten(sublist):
                yield element
    except TypeError:
        yield nested


def main():
    list1 = [[1.3, 2, 3], [5.0, 6], [4], [9, 5, 4, 3]]
    result1 = list(flatten(list1))
    print(result1)

    list2 = [[1], [1, [3], [5, [6]]]]
    result2 = list(flatten(list2))
    print(result2)


if __name__ == '__main__':
    main()