牛客题解 | 矩阵转置

题目

题目链接

对于给定的矩阵 A，其转置矩阵表示为 A^T。数学表达式为：

$$A_{ij}^T=A_{ji}$$
其中，$A_{ij}^T$ 是转置矩阵 A^T 的第 i 行第 j 列元素，$A_{ji}$ 是原矩阵 A 的第 j 行第 i 列元素。标准代码如下：

def transpose_matrix(a: List[List[Union[int, float]]]) -> List[List[Union[int, float]]]:
    return [list(i) for i in zip(*a)]

当然也可以使用numpy库的.T方法简化计算

def transpose_matrix(a: List[List[Union[int, float]]]) -> List[List[Union[int, float]]]:
    import numpy as np
    return np.array(a).T.tolist()