我想在Python中创建一个矩阵转置功能。 矩阵是一个二维阵列,表示为整数的列表的列表。 例如,下面是2X3矩阵(意味着矩阵的高度为2,宽度为3):
A=[[1, 2, 3],
[4, 5, 6]]
要在第i个指标被调换了第j个项目应成为第j个指标的第i个项目。 下面是上面的示例看起来换位:
>>> transpose([[1, 2, 3],
[4, 5, 6]])
[[1, 4],
[2, 5],
[3, 6]]
>>> transpose([[1, 2],
[3, 4]])
[[1, 3],
[2, 4]]
我怎样才能做到这一点?
您可以使用zip
与*
获得矩阵转置的:
>>> A = [[ 1, 2, 3],[ 4, 5, 6]]
>>> zip(*A)
[(1, 4), (2, 5), (3, 6)]
>>> lis = [[1,2,3],
... [4,5,6],
... [7,8,9]]
>>> zip(*lis)
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
如果你想在返回的列表是列表的列表:
>>> [list(x) for x in zip(*lis)]
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
#or
>>> map(list, zip(*lis))
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
有没有偷懒,并使用与NumPy阵列的置函数的奖品? ;)
import numpy as np
a = np.array([(1,2,3), (4,5,6)])
b = a.transpose()
如果我们想返回相同的矩阵,我们可以这样写:
return [[ m[row][col] for col in range(0,width) ] for row in range(0,height) ]
这里做的事情是通过每个行会和每列返回每个元素遍历一个矩阵m。 所以为了将这样的:
[[1,2,3],
[4,5,6],
[7,8,9]]
现在对于问题3,我们不是要通过列去列,每一行中返回的每一个元素。 所以为了将这样的:
[[1,4,7],
[2,5,8],
[3,6,9]]
因此只需切换中,我们反复顺序:
return [[ m[row][col] for row in range(0,height) ] for col in range(0,width) ]