题解 | 矩阵变换

import numpy as np
def transform_matrix(A,T,S):
    # 补全代码
    an=np.array(A,dtype=np.float64)
    tn=np.array(T,dtype=np.float64)
    sn=np.array(S,dtype=np.float64)
    detsn=np.linalg.det(sn)
    dettn=np.linalg.det(tn)
    if np.isclose(detsn,0) or np.isclose(dettn,0):
        return -1
    invtn=np.linalg.inv(tn)
    res=invtn@an@sn
    return np.round(res,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)