Jacobi Method Python Numpy