python矩阵转置_867. 矩阵转置（Python）

题目

难度：★☆☆☆☆

类型：几何、二维数组、数学

给定一个矩阵 A， 返回 A 的转置矩阵。

矩阵的转置是指将矩阵的主对角线翻转，交换矩阵的行索引与列索引。

示例

示例 1

输入：[[1,2,3],[4,5,6],[7,8,9]]

输出：[[1,4,7],[2,5,8],[3,6,9]]

示例 2

输入：[[1,2,3],[4,5,6]]

输出：[[1,4],[2,5],[3,6]]

提示

1 <= A.length <= 1000

1 <= A[0].length <= 1000

解答

转置前矩阵的维度是r=len(A), c=len(A[0])，转置后矩阵的维度应该交换，首先我们构建转置后的矩阵，并填充所有值为空，然后遍历A矩阵中的每一个点，把它放在B上对应的位置即可：B[j][i]=A[i][j]。

class Solution:

def transpose(self, A):

"""

:param A: List[List[int]]

:return: List[List[int]]

"""

B = [[None for _ in range(len(A))] for _ in range(len(A[0]))]

for i in range(len(A)):

for j in range(len(A[0])):

B[j][i] = A[i][j]

return B

在python中有zip方法，可以实现快速的矩阵转置：

class Solution:

def transpose(self, A):

"""

:param A: List[List[int]]

:return: List[List[int]]

"""

return list(zip(*A))

如有疑问或建议，欢迎评论区留言~