curve fit scatter plot python