Python将多维列表「拉伸」为一维列表的10种方式
来源:投稿 作者:Fairy
编辑:学姐
在Python编程中,列表是一种常用的数据类型。当我们遇到了一个嵌套列表,如果想将它扁平化为一维列表,就可以使用下面10种方法之一来实现这个需求。
1. 使用两层循环遍历
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。然后创建了一个空列表new_lst
。接下来使用for
循环遍历lst中的每一个子列表,再使用嵌套的for循环遍历子列表中的每一个元素,并将这些元素添加到new_lst中。最后输出new_lst,即为将lst扁平化为一维列表的结果。

2. 使用flatten()函数
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。接下来,使用NumPy
库的array()
方法将lst转换成了一个NumPy数组,再调用该数组的flatten()
方法将其扁平化为一个一维数组。最后,将该一维数组转换成一个标准的Python
列表,并将其赋值给变量new_lst
。最终输出new_lst,即为将lst扁平化为一维列表的结果。

3. 使用列表解析式
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。然后,使用一行代码创建了一个新的一维列表new_lst
。具体来说,这行代码中使用了两层for
循环,并在内部使用了列表推导式的语法。在这里,外层for循环用于遍历lst中的每一个子列表,而内层for循环则用于遍历子列表中的每一个元素。最后,在列表推导式中使用了element
这个变量来存储遍历到的元素,并将其添加到新列表new_lst中。

4. 使用reduce()函数
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。接下来,使用reduce()
函数和operator.concat
函数实现将lst中的所有子列表连接在一起。具体来说,operator.concat函数用于将两个列表连接在一起,而reduce()函数则用于对lst中的每一个子列表依次执行concat
操作。最终,reduce函数得到的结果是一个一维列表,存储在变量new_lst中。

5. 使用from_iterable()函数
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。接下来,使用itertools.chain.from_iterable()
函数实现将lst中的所有子列表连接在一起。具体来说,from_iterable()
函数用于将一个可迭代对象中的所有元素连接到一起,而chain()
函数则用于在多个可迭代对象之间建立链接。最终,使用list()
函数将生成的迭代器转换为一个新的一维列表new_lst
。

6. 使用sum()函数
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。接下来,使用sum()
函数实现将lst中的所有子列表连接在一起。具体来说,sum()函数用于对多个可迭代对象进行求和操作,而对于列表而言,在使用sum()函数时传入一个空列表[]作为初始值,就可以将多个子列表拼接在一起。最终,使用list()函数将生成的结果转换为一个新的一维列表new_lst。

7. 使用ravel()函数
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。接下来,使用numpy
中的ravel
函数对lst进行扁平化操作,并将其转换为一个一维数组。最后,使用list()函数将numpy数组转换为一个新的一维列表new_lst。

8. 使用reshape()函数
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。接下来,使用numpy.array()
函数将lst转换为一个numpy数组,并使用reshape()
函数对其进行重新形状,将其扁平化为一个一维数组。最后,使用tolist()
函数将numpy数组转换为一个新的一维列表new_lst。

9. 使用flatten()函数
这段代码中,定义了一个二维列表lst
,其中包含了三个子列表(即嵌套的列表)。接下来,使用numpy.array()
函数将lst转换为一个numpy数组,并使用flatten()
函数对其进行扁平化操作,将其扁平化为一个一维数组。需要注意的是,该函数传入的参数'C'
表示扁平化时按照行优先顺序进行,也就是将每一行展开到一起。最后,使用list()函数将numpy数组转换为一个新的一维列表new_lst。

10. 使用stack()函数
这段代码中,定义了一个二维列表lst,其中包含了三个子列表(即嵌套的列表)。接下来,使用pandas.DataFrame()函数将lst转换为一个DataFrame数据框,并使用stack()函数对其进行堆叠操作,将其扁平化为一个Series数据序列。然后,使用list()函数将Series转换为一个新的一维列表new_lst。

以上10种方式,你了解哪几种呢?
推荐课程
《Python · AI&数据科学入门》

关注“学姐带你玩AI”公众号,回复“Python”了解课程详情~