Simple linear regression

linearRegression.py

# https://docs.scipy.org/doc/scipy/reference/generated/scipy.stats.linregress.html
import numpy as np
from scipy import stats
import matplotlib.pyplot as plt

x = np.array([1,2,3])
linregress(np.arange(len(x)), x)
slope, intercept, r_value, p_value, std_err = stats.linregress(np.arange(len(x)), x)

#plot
plt.plot(x, y, 'o', label='original data')

plt.plot(x, intercept + slope*x, 'r', label='fitted line')

plt.legend()

plt.show()