Linear Extrapolation Python