Gauss seidel method python numpy