python 二维列表按行列取、降成一维列表

Python将二维/多维数组转换为一维
发表时间：2020-06-26
Python将二维数组/多维数组转换为一维
方法1:flatten
方法2:reshape+concatenate
方法3:sum()
方法4:列表推导式
方法5:operator
方法6:itertools
方法1:flatten
import numpy as np
mulArrays = [[1,2,3],[4,5,6],[7,8,9]]
print(list(np.array(mulArrays).flatten()))
方法2:reshape+concatenate
import numpy as np
mulArrays = [[1,2,3],[4,5,6],[7,8,9]]
print(list(np.concatenate(array.reshape((-1,1),order="F"))))
方法3:sum()
mulArrays = [[1,2,3],[4,5,6],[7,8,9]]
print(sum(mulArrays,[])) #[1, 2, 3, 4, 5, 6, 7, 8, 9]
方法4:列表推导式
mulArrays = [[1,2,3],[4,5,6],[7,8,9]]
print([i for arr in mulArrays for i in arr]) #[1, 2, 3, 4, 5, 6, 7, 8, 9]
方法5:operator
import operator
from functools import reduce
mulArrays = [[1,2,3],[4,5,6],[7,8,9]]
print(reduce(operator.add, mulArrays))
方法6:itertools
from itertools import chain
mulArrays = [[1,2,3],[4,5,6],[7,8,9]]
print(list(chain.from_iterable(mulArrays))) #[1, 2, 3, 4, 5, 6, 7, 8, 9]

文章目录
按行取
最简方法
使用 numpy
按列取
列表推导式
map 函数
使用 numpy
降成一维
解包操作
解包拓展
itertools.chain
使用 numpy

以简单的 2*2 二维列表为例
a = [[1,2], [3,4]]

按行取
最简方法
i = 0 # 取第一行
a[i] # 结果：[1, 2]

使用 numpy
i = 0 # 取第一行
b = np.array(a)
b[i,:] # 也等价于 b[i]，结果：array([1, 2])

按列取
列表推导式
i = 0 # 取第一列
[item[i] for item in a] # 结果：[1, 3]

map 函数
i = 0 # 取第一列
list(map(lambda x: x[i], a)) # 结果：[1, 3]
1
2
使用 numpy
i = 0 # 取第一列
b = np.array(a)
b[:, i] # 结果：array([1, 3])

降成一维
解包操作
b, c = a # b==[1, 2], c==[3, 4]
b+c # 结果：[1, 2, 3, 4]

解包拓展
b, c = a # b==[1, 2], c==[3, 4]
[*b, *c] # py3.5版本之后解包支持作用在表达式中，结果：[1, 2, 3, 4]

itertools.chain
from itertools import chain 
list(chain(*a)) # 或者 list(chain.from_iterable(a))，结果：[1, 2, 3, 4]

使用 numpy
b = np.array(a)
b.flatten() # 结果：array([1, 2, 3, 4])