import warnings warnings.filterwarnings(\'ignore\', \'The iteration is not making good progress\') import matplotlib.pyplot as plt import numpy as np import math imp