Question

我正在尝试在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]]

我该怎么做？

Answer 1

您可以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]]

Answer 2

是否有懒惰和使用NumPy阵列的转置功能的奖品？ ;）

import numpy as np

a = np.array([(1,2,3), (4,5,6)])

b = a.transpose()

Answer 3

如果我们想要返回相同的矩阵，我们会写：

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) ]

在Python中转置矩阵

3 个答案: