Python code for gauss seidel method