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

import numpy as np

def gauss_seidel(A, b, n, x_ini=None):
    x = x_ini or np.zeros_like(b)
    dim = len(b)
    for _ in range(n):
        for i in range(dim):
            sum_front = np.dot(A[i, :i], x[:i])
            sum_back = np.dot(A[i, i+1:], x[i+1:])
            x[i] = (b[i] - sum_front - sum_back) / A[i, i]
    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())