牛客题解 | 矩阵转置

题目

题目链接

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

\[A^T_{ij} = A_{ji} \]

其中，\(A^T_{ij}\) 是转置矩阵 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()