在python中实现Gauss seidel

def gauss_seidel(A, b, x0, max_iter, tol):
    n = len(A)
    x = x0.copy()
    iter_count = 0
    error = tol + 1

    while iter_count < max_iter and error > tol:
        x_prev = x.copy()
        for i in range(n):
            sum1 = sum(A[i][j] * x[j] for j in range(i))
            sum2 = sum(A[i][j] * x_prev[j] for j in range(i + 1, n))
            x[i] = (b[i] - sum1 - sum2) / A[i][i]

        error = max(abs(x[i] - x_prev[i]) for i in range(n))
        iter_count += 1

    return x