题解 | 使用Jacobi方法求解线性方程组

import numpy as np
def solve_jacobi(A, b, n) :
    # 补全代码
    a=np.array(A,dtype=np.float64)
    b=np.array(b,dtype=np.float64)
    size=len(b)
    x=np.zeros(size)
    d=np.diag(np.diag(a))
    d_inv=np.diag(1/np.diag(a))
    lu=a-d
    for _ in range(n):
        x=d_inv@(b-lu@x)
    return np.round(x,4).tolist()
# 主程序
if __name__ == "__main__":
    # 输入
    ndarrayA = input()
    ndarrayB = input()
    n = input()

    # 处理输入
    import ast
    A = ast.literal_eval(ndarrayA)
    b = ast.literal_eval(ndarrayB)
    n = int(n)

    # 调用函数计算
    output = solve_jacobi(A,b,n)
    
    # 输出结果
    print(output)