一元线性回归Python实现

#Monadic linear regression
#《概率论与数理统计》797页例8.4.1
from matplotlib import pyplot as plt
import numpy as np

x=np.arange(0.1,0.24,0.01)
x=np.around(x,decimals=2)
index1=np.where(x==0.19)
index2=np.where(x==0.22)
x=np.delete(x,(index1,index2))
y=np.array([42,43,45,45,45,47.5,49,53,50,55,55,60])
plt.scatter(x,y,color="r")#散点图

lxx=np.sum((x-np.mean(x))**2)
lxy=np.sum((x-np.mean(x))*(y-np.mean(y)))
b1=lxy/lxx
b0=np.mean(y)-b1*np.mean(x)
x_bar=np.arange(0.075,0.25,0.01)
y_bar=b0+b1*x_bar
plt.plot(x_bar,y_bar)