Python Linear Fitting Equation To Data