牛客题解 | 矩阵变换

题目

题目链接

矩阵变换是一种将矩阵进行变换的方法，其计算公式为：

$$A=T^{-1}\times A\times S$$
其中，$A$ 是输入矩阵，$T$ 和 $S$ 是变换矩阵，并且$T$和$S$皆为逆矩阵。

标准代码如下

def transform_matrix(A,T,S):
    A = np.array(A, dtype=float)
    T = np.array(T, dtype=float)
    S = np.array(S, dtype=float)
    
    if np.linalg.det(T) == 0 or np.linalg.det(S) == 0:
        return -1
    
    T_inv = np.linalg.inv(T)

    transformed_matrix = np.round(T_inv @ A @ S, 3)
    
    return transformed_matrix.tolist()