题解 | 矩阵变换

import numpy as np
def transform_matrix(A,T,S):
    # 补全代码
    if np.linalg.det(T) == 0 or np.linalg.det(S) == 0:
        return -1
    T_inv_A = np.dot(np.linalg.inv(T), A)
    return np.round(np.dot(T_inv_A, S), 3).tolist()

# 主程序
if __name__ == "__main__":
    # 输入
    ndarrayA = input()
    ndarrayT = input()
    ndarrayS = input()

    # 处理输入
    import ast
    A = ast.literal_eval(ndarrayA)
    T = ast.literal_eval(ndarrayT)
    S = ast.literal_eval(ndarrayS)

    # 调用函数计算
    output = transform_matrix(A,T,S)
    
    # 输出结果
    print(output)