Linear Fit Python Numpy