python list元素展开

最新推荐文章于 2024-10-25 18:36:44 发布

冰山一角的岁月神偷

最新推荐文章于 2024-10-25 18:36:44 发布

阅读量4.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： python语言文章标签： python

本文链接：https://blog.youkuaiyun.com/u012279165/article/details/77771367

python语言专栏收录该内容

9 篇文章

订阅专栏

本文介绍如何利用Python的itertools模块中的chain.from_iterable()函数来将嵌套列表进行扁平化处理，适用于单层或多层相同结构的嵌套情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

单层嵌套

import itertools
a = [[1,2,3],[4,5,6], [7], [8,9]]
out = list(itertools.chain.from_iterable(a))#如果有嵌套的化需要多次调用此函数

输出：[1, 2, 3, 4, 5, 6, 7, 8, 9]
当遇到多层嵌套(要求嵌套的结构必须相同)时只要重复调用上述函数即可

import itertools
a = [[[1,1],[2,2],[3,3]],[[4,4],[5,5],[6,6]], [[7,7]], [[8,8],[9,9]]]
out = list(itertools.chain.from_iterable(a))
#输出：[[1, 1], [2, 2], [3, 3], [4, 4], [5, 5], [6, 6], [7, 7], [8, 8], [9, 9]]
out_new= list(itertools.chain.from_iterable(out))
#输出:[1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9]