回归预测房价

y=wx+b

 

import matplotlib.pyplot as plt
x=[123,150,87,102]
y=[250,320,160,220]
plt.scatter(x, y,c='r',marker='*',s=100) #c:设置散点的颜色 marker设置散点形状
xmean = sum(x)/len(x)
ymean = sum(y)/len(y)
s1 = 0
s2 = 0
for i in range(0, len(x)):
    s2 = s2 + (x[i]-xmean)*(x[i]-xmean)
    s1 = s1 + (x[i]-xmean)*(y[i]-ymean)
w = s1/s2
b = ymean-w*xmean
x_true = 75                                       #修改这里可以实现预测
y_pred = x_true*w + b
print(y_pred)                                     #输出预测值

推导可得：

标题