Python--线性回归

线性回归

线性回归分为简单一维和多维,由于就为而言极大可能用不到多维的,这里就只记录一维。

numpy.polyfit(x,y,degree,rcond=None,full=False,w=None,cov=False)
  • x,y:原始数据
  • degree:维度
  • 返回值是ax+b,的系数[a,b];如果是多维的话就是返回系数矩阵
from pylab import *
import sympy as sy 
x=[1,2,3,4,5]
y=[1.1,2.1,3,4,5]
scatter(x,y)
[a,b]=np.polyfit(x,y,1)
x=np.linspace(0,6,100)
plot(a*x+b,x)
text(1.5,3,'y={:.2f}x+{:.2f}'.format(a,b))
xlabel('x')
ylabel('y')
Text(0, 0.5, 'y')
png