如何在Python中展平浅层列表?


展平浅层列表意味着将嵌套列表转换为简单的单维度列表。换句话说,就是将多维列表转换为一维列表。

展平过程可以使用不同的技术来执行,例如嵌套for循环、列表推导式、列表连接以及使用内置函数。

在本文中,我们将讨论几种展平Python浅层列表的技术。

使用嵌套for循环

通过使用嵌套for循环和list.append()方法,我们可以展平浅层列表。让我们来看一下如何在程序中实现这一点。

示例

这个简单的例子成功地使用嵌套for循环展平了一个浅层列表。第一个循环用于迭代列表中的子列表,第二个循环用于迭代子列表中的项目。

li = [[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]]
print("Original shallow list:", li)

flatlist = []   
for sub_li in li:
    for ele in sub_li:
        flatlist.append(ele)
print("After flattening :", flatlist)

输出

Original shallow list: [[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]]
After flattening : [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]

使用列表推导式

此技术的工作原理与嵌套for循环非常相似,就像上面的解决方案一样,它使用列表推导式打包成一行代码。

列表推导式是一种基于现有可迭代对象(列表、字符串、元组)创建列表的简单而优雅的方法。

语法

newList = [ expression(element) for element in oldList if condition ]

示例

我们使用列表推导式方法在一行代码中展平了一个浅层列表。

li = [[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]]
print("Original shallow list:", li)

flatlist = [ele for ele in sub_li for sub_li in li]   

print("List after flattening :", flatlist)

输出

Original shallow list: [[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]]
List after flattening : [90, 90, 90, 100, 100, 100, 110, 110, 110, 120, 120, 120]

使用itertools (chain())

chain()是一种接受一系列可迭代对象然后返回单个可迭代对象的方法,该可迭代对象不能直接使用,必须显式转换。要访问chain()方法,我们需要首先将itertools模块导入到我们的程序中。

语法

chain (*iterables)

示例

为了解压嵌套列表,我们使用了*运算符。

import itertools

li = [[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]]
print("Original shallow list:", li)

flatlist = list(itertools.chain(*li))
print("List after flattening :", flatlist)

输出

Original shallow list: [[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]]
List after flattening : [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]

使用sum()函数

对内部列表(子列表)求和是另一种解决方案。在这里,我们将使用Python内置函数sum()。

语法

sum(iterable, start)

示例

在这个例子中,参数iterable接受一个列表的列表,start是一个空列表,它作为初始的扁平列表,内部子列表的项目将被添加到其中。

li = [[1, 2, 3, 4], [5, 6, 7], [8, 9]]
print("Original shallow list:", li)

faltlist = sum(li, [])
print("List after flattening :", faltlist)

输出

Original shallow list: [[1, 2, 3, 4], [5, 6, 7], [8, 9]]
List after flattening : [1, 2, 3, 4, 5, 6, 7, 8, 9]

我们已经讨论了几种展平Python浅层列表的技术。

更新于:2023年8月24日

1K+ 次浏览

启动你的职业生涯

完成课程获得认证

开始学习
广告
© . All rights reserved.