Curve fitting with python

curvefit.py

import matplotlib 
matplotlib.use('agg')
import matplotlib.pyplot as plt
import numpy as np
from scipy.optimize import curve_fit

x = np.linspace(0,100)
noise = np.random.normal(0, 2, len(x))
simulated_data  = x*3 + 5 + noise

def func(x, m, b):
    return m*x + b

params = curve_fit(func, x, simulated_data)
m, b = params[0]
print("m=", m, " b=", b)

plt.plot(x, m*x + b, label="fit")
plt.plot(x, simulated_data, label="sim data", alpha=0.5)
plt.legend()
plt.savefig("test.png")