最小二乘法曲線擬合+C代碼

最小二乘法曲線的係數求解過程是解一個正規方程組的解的過程，下圖是數值分析課本上，最小二乘法擬合的原理：

課本中的例子如下：

c代碼如下：

#include "stdio.h"
#include "stdlib.h"
#include "math.h"
#include "vector"
using namespace std;


struct point
{
double x;
double y;
};


typedef vector<double> doubleVector;


vector<point> getFileInf(char *File);  //獲取文件數據
doubleVector getCoeff(vector<point> sample, int n);   //矩陣方程


void main()
{
int i, n;
char *File = "樣本.txt";
vector<point> sample;
doubleVector Coef;


sample = getFileInf(File);


printf("擬合階數n：");
scanf("%d", &n);


Coef = getCoeff(sample, n);


printf("\n擬合矩陣的係數爲：\n");
for(i=0; i<Coef.size(); i++)
printf("a%d = %lf\n", i, Coef[i]);


}




//矩陣方程
doubleVector getCoeff(vector<point> sample, int n)
{
vector<doubleVector> matFunX;  //矩陣方程
vector<doubleVector> matFunY;  //矩陣方程
doubleVector temp;
double sum;
int i, j, k;


//正規方程X
for(i=0; i<=n; i++)
{
temp.clear();
for(j=0; j<=n; j++)
{
sum = 0;
for(k=0; k<sample.size(); k++)
sum += pow(sample[k].x, j+i);
temp.push_back(sum);
}

matFunX.push_back(temp);
}




//正規方程Y
for(i=0; i<=n; i++)
{
temp.clear();
sum = 0;
for(k=0; k<sample.size(); k++)
sum += sample[k].y*pow(sample[k].x, i);
temp.push_back(sum);

matFunY.push_back(temp);
}




//矩陣行列式變換
double num1, num2, ratio;


for(i=0; i<matFunX.size()-1; i++)
{
num1 = matFunX[i][i];
for(j=i+1; j<matFunX.size(); j++)
{
num2 = matFunX[j][i];
ratio = num2/num1;


for(k=0; k<matFunX.size(); k++)
matFunX[j][k] = matFunX[j][k]-matFunX[i][k]*ratio;


matFunY[j][0] = matFunY[j][0]-matFunY[i][0]*ratio;
}


}




//計算擬合曲線的係數
doubleVector coeff(matFunX.size(), 0);
for(i=matFunX.size()-1; i>=0; i--)
{
if(i==matFunX.size()-1)
coeff[i] = matFunY[i][0]/matFunX[i][i];

else
{
for(j=i+1; j<matFunX.size(); j++)
matFunY[i][0] = matFunY[i][0]-coeff[j]*matFunX[i][j];
coeff[i] = matFunY[i][0]/matFunX[i][i];
}
}




return coeff;
}




//獲取文件數據
vector<point> getFileInf(char *File)
{
int i=1;
vector<point> dst;


FILE *fp = fopen(File, "r");


if(fp==NULL)
{
printf("Open file error!!!\n");
exit(0);
}


point temp;
double num;


while(fscanf(fp, "%lf", &num)!=EOF)
{
if(i%2==0)
{
temp.y = num;
dst.push_back(temp);
}
else
temp.x = num;
i++;
}


fclose(fp);


return dst;
}

運行結果如下：