867. 转置矩阵

给定一个矩阵 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. 1 <= A.length <= 1000
2. 1 <= A[0].length <= 1000

class Solution:
    def transpose(self, A: 'List[List[int]]') -> 'List[List[int]]':
        # 交换矩阵的行索引与列索引
        res = [[0] * len(A) for i in range(len(A[0]))]
        for r, row in enumerate(A):
            for c, val in enumerate(row):
                res[c][r] = val
        return res

    def transpose2(self, A: 'List[List[int]]') -> 'List[List[int]]':
        return [list(value) for value in zip(*A)]

if __name__=='__main__':
    s = Solution()
    # A = [[1,2,3],[4,5,6],[7,8,9]]
    A = [[1,2,3],[4,5,6]]
    print(s.transpose(A))