在Python中转置矩阵[重复]答案

【问题标题】：Transpose a matrix in Python [duplicate]在Python中转置矩阵[重复]
【发布时间】：2013-06-06 21:59:54
【问题描述】：

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

我该怎么做？

【问题讨论】：

你可以在这里找到答案：stackoverflow.com/questions/6473679/…

标签： python list list-comprehension

【解决方案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]]

【讨论】：

在 Python 3 中，map 返回一个迭代器，因此要从最后一个解决方案中获取列表，您需要 list(map(list, zip(*lis))) 或 [*map(list, zip(*lis))]。
你能解释一下zip(*lis)中的*是如何工作的吗

【解决方案2】：

懒惰使用 NumPy 数组的转置函数有奖励吗？ ;)

import numpy as np

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

b = a.transpose()

【讨论】：

由于给定的骨架函数，您的答案不正确。这个答案必须使用列表理解来完成。下面的答案是正确的。不过信息很好。
您不仅要导入一个庞大的库来做一些可以用一行纯 python 实现的事情，而且您的答案在技术上也是错误的，因为输出不是列表中所示的列表问题，但是一个numpy数组。所以答案更长更慢，因为你要转换成numpy数组，转置，再转换回来。

【解决方案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) ]

【讨论】：