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

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