以简单的 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]
使用 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])