Python基础 - 将二维列表转换为一维-Toy模板网

这篇具有很好参考价值的文章主要介绍了Python基础 - 将二维列表转换为一维。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在实验中经常会遇到将二维列表(数组)拉平到一维，如将 [[1, 1], [2, 2]] 转换为[1, 1, 2, 2]，有以下几种操作方案：

1. 最简单的直接使用循环，如下：

list1 = [[1, 1], [2, 2]]
ans = [item for list in list1 for item in list]
print(ans) # [1, 1, 2, 2]

2. 使用itertools.chain(*iterables)，能够去除iterable里的内嵌的一层iterable【注意：只能去除一层，多的层数去除不了，具体实例可看下面】，其中iterables为可迭代对象，如list, string, dict。

from itertools import chain
list1 = [[1, 1], [2, 2]]
ans = list(chain.from_iterable(list1))
print(ans) # [1, 1, 2, 2]


list2 = [[[1, 1]], [2, 2]]
ans2 = list(chain.from_iterable(list2))
print(ans2) # [[1, 1], 2, 2]

3. 使用a.flatten()，但是前提是要把list转换成numpy格式。【注意：无论array外面套了多少层，flatten()都会将其展平成一维。

import numpy as np
list1 = [[1, 1], [2, 2]]
list1 = np.array(list1)
ans = list1.flatten()
ans = ans.tolist()

4. sum(二维列表, [])函数小技巧，但这种方式也只能展开最外面一层，跟itertools.chain()类似。

list1 = [[1, 1], [2, 2]]
print(sum(list1, []))

为什么这种方式可以展平呢？我们来具体看下：

(1) 首先我们要知道在python中两个列表相加为多少？

[1, 2, 3] + [4, 5, 6] = [1, 2, 3, 4, 5, 6]。没错列表相加等价于列表拼接。有同学肯定会想，这个操作跟extend是一样的，其实不同extend是在原对象的基础上进行添加元素，而+操作则会新建一对象

(2) 其次我们把sum拆解来看。sum的本质就是列表内元素相加，所以list1中的[1, 1], [2, 2]就会相加，得到[1, 1, 2, 2]，然后得到的值又会和后面的空列表[]进行一次相加操作，也就是[1, 1, 2, 2] + [] = [1, 1, 2, 2]，没变。所以会得到展开后的列表。

(3) 那sum后面能不能不放这个空列表[]呢？肯定不行。如果不放，则会默认[1, 1, 2, 2]和0相加，会报错，如下：

list1 = [[1, 1], [2, 2]]
print(sum(list1))
Traceback (most recent call last):
  File "<input>", line 1, in <module>
TypeError: unsupported operand type(s) for +: 'int' and 'list'

相似博文将子列表相同位置的元素重组，指路：Python基础 - 将子列表相同位置的元素重组文章来源地址https://www.toymoban.com/news/detail-455955.html

到了这里，关于Python基础 - 将二维列表转换为一维的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！