СЛАУ методом простых итераций

import numpy as np
 
def func(a, b, maxi, acc):
    iter = 0
    x0 = np.zeros(len(b))
    for k in range(maxi):
        x1 = np.zeros(len(b))
        for i in range(len(b)):
            s = 0
            for j in range(len(b)):
                if j != i:
                    s += a[i,j] * x0[j]
            x1[i] = (b[i] - s) / a[i,i]
        iter += 1
        if np.allclose(x0, x1, acc):
            break
        x0 = x1
    print('Число итераций:', iter)
    return x1
 
a1 = np.array([[10.,2.,0.],[1.,5.,1.],[2.,3.,10.]])
b1 = np.array([7.,-8.,6.])
 
t = 1e-10 #точность
mi = 100 # max итераций
 
print(func(a1, b1, mi, t))

import numpy as np
 
def sort_diag_arr(a,b) :
    n = len(a)
    a = np.array(a)
    b = np.array(b)
    for i in range(n):
        ind = np.abs(a[i:, i:n]).argmax()
        row = ind // (n - i) + i
        col = ind % (n - i) + i 
        a[:,[i, col]] = a[:,[col, i]]
        a[[i, row], :] = a[[row, i], :]
        b[i], b[row] = b[row], b[i]
    print(a)
    return a, b
 
 
A = np.array([[10.,2.,0.],[1.,5.,1.],[2.,3.,10.]])
B = np.array([7.,-8.,6.])
eps = 1e-3
 
a, b = sort_diag_arr(A,B)
diag = (1/np.diag(a)).reshape(-1,1)
a[np.diag_indices_from(a)] = 0.
a = np.hstack((-a, b.reshape(-1, 1))) * diag
x = a[:,-1].ravel()
x = np.append(x, 1)
 
tmp = x.copy() + eps
cnt = 0
while abs(x-tmp).sum() > eps:
    tmp = x.copy()
    x = a.dot(x)
    x = np.append(x, 1)
    cnt += 1
 
print(x[:-1].round(3), cnt)