题解 | Gauss-Seidel法求解线性方程组

import numpy as np

def gauss_seidel(A, b, n, x_ini=None):
    X=np.zeros(len(b))
    l=A.shape[0]
    h=A.shape[1]
    for k in range(n):
        X_new=np.zeros(h)
        for i in range(h):
            z_sum=X_new[:i]@A[i][:i]
            p_sum=X[i+1:]@A[i][i+1:]
            X_new[i]=(b[i]-p_sum-z_sum)/A[i][i]
        X=X_new
            
    return X

if __name__ == "__main__":
    A = np.array(eval(input()), dtype=float)
    b = np.array(eval(input()), dtype=float)
    n = int(input())
    print(gauss_seidel(A, b, n).tolist())