题解|矩阵转置

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

其中， 是转置矩阵 A^T 的第 i 行第 j 列元素， 是原矩阵 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()