Python 1d Line Fit