Решение СЛАУ методом Якоби

# -*- coding: utf-8 -*-
import numpy as np
from math import fabs
 
 
matrix = np.array([[-4, 0.5, -1.5, -1],
                   [0.55, 6, 2.4, 1.3],
                   [0.33, -0.52, 3, 0.21],
                   [-0.22, 0.98, -2.5, 5]])
B = np.array([[-1.67],
              [3.15],
              [-0.21],
              [5.54]])
X = np.array([[0.0],
              [0.0],
              [0.0],
              [0.0]])
deltaX = np.array([[0.0],
                   [0.0],
                   [0.0],
                   [0.0]])
 
 
def chek(i):
    j = 0
    sum = 0
    for j in range (0, 4, 1):
        if i != j:
            sum = sum + fabs(matrix[i,j])
    if fabs(matrix[i,i]) > sum:
        if i != 3:
            return chek(i+1)
        if i == 3:
            print (u"Система збіжна по Якобі")
    else:
        print(u"Помилка, система не збіжна")
        return
 
 
chek(0)
eps = 0.0001
 
for i in range (0,4,1):
    B[i,0] = B[i,0]/matrix[i,i]
 
i = 0
j = 0
for i in range (0,4,1):
    t = float(matrix[i,i])
    for j in range (0,4,1):
        matrix[i,j] = matrix[i,j]/t
        matrix[i,j] = matrix[i,j]*-1
    matrix[i][i] = 0
print matrix
print B
print ("--------------------------------------------------------------------------------------------------------------")
 
counter = 0
max = 1.0000
while max > eps:
    counter +=1
    if counter > 1:
        B = X[:]
    print B
    for i in range(0,4,1):
        SUM = 0
        s = 0
        for j in range(0,4,1):
            s = matrix[i,j]*B[j,0]
            SUM = SUM+s
        SUM = SUM+B[i,0]
        X[i,0] = SUM
        deltaX[i, 0] = fabs((B[i, 0] - SUM))
    print (counter, "Iteraciya")
    print ("X=" ,X)
    print ("deltaX=", deltaX)
    max = deltaX[0,0]
    for k in range(0,4,1):
        if deltaX[k,0] > max:
            max = deltaX[k,0]
print ("--------------------------------------------------------------------------------------------------------------")
 
print u"Вектор розв'язкiв"
print X
print deltaX