在Python中转置矩阵

参见英文答案 > Transpose list of lists                                    10个
我正在尝试在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]]

转载注明原文:在Python中转置矩阵 - 代码日志