Python实现线性拟合

线性方程 ax+b的Python代码实现

import matplotlib.pyplot as plt
import numpy as np
#y = a*x^2 + b*x + c
def LeastSquares( x,  y,  data_num):
   sumx=0
   sumx2=0
   sumy=0
   sumxy=0
   D=0

   for i in range(0,data_num,1):
      sumx+=x[i]
      sumy+=y[i]
      sumx2+=x[i]**2
      sumxy+=x[i]*y[i]
   k =((data_num*sumxy)-(sumx*sumy))/((data_num*sumx2)-(sumx*sumx))
   b=((sumx2*sumy)-(sumx*sumxy))/((data_num*sumx2)-(sumx*sumx))

   return k,b;

if __name__=="__main__":

  
   '''训练数据'''
   x = [0, 2, 4, 6, 8, 10]
   y = [0