本文介绍了几种在Python中将二维列表转换为一维列表的方法，包括使用列表推导式、sum函数以及itertools.chain函数。这些技巧对于处理复杂的数据结构非常有用。
方法一：使用列表推导式
>>> vec = [[1,2,3],[4,5,6],[7,8,9]]
>>> get = [num for elem in vec for num in elem]
>>> get
[1, 2, 3, 4, 5, 6, 7, 8, 9]
方法相当于
>>> vec = [[1,2,3],[4,5,6],[7,8,9]]
>>> result = []
>>> for elem in vec:
for num in elem:
result.append(num)
>>> result
[1, 2, 3, 4, 5, 6, 7, 8, 9]
方法二：
使用sum函数
>>> vec = [[1,2,3],[4,5,6],[7,8,9]]
>>> get = sum(vec,[])
>>> get
[1, 2, 3, 4, 5, 6, 7, 8, 9]
或者是使用chain函数
>>> vec = [[1,2,3],[4,5,6],[7,8,9]]
>>> from itertools import chain
>>> list(chain(*vec))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
扫一扫
1021
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
