用python实现简单线性回归

import numpy as npy
def fitSLR(x,y):
fenzi = 0
fenmu = 0
num=len(x)
for i in range(num):
fenzi=fenzi+(x[i]-npy.mean(x))*(y[i]-npy.mean(y))
fenmu=fenmu+(x[i]-npy.mean(x))**2
b1=fenzi/float(fenmu)
b0=npy.mean(y)-b1*npy.mean(x)
print(b1,b0)
return b0,b1
def predict(text_x,b1,b0):
return b1*text_x+b0
x=[1,3,2,1,3]
y=[14,24,18,17,27]
b0,b1=fitSLR(x,y)
rst=predict(6,b1,b0)
print(rst)