最小二乘法直线拟合的C语言及matlab实现
这里直接给代码,理论推导请参看最小二乘法的理论推导
- C语言实现
#include <stdio.h>
//最小二乘法求散点拟合函数y=a+bx的参数a和b
void LFM(double x[],double y[],int n)
{
int i;
double a=0,b=0;
double sum1=0,sum2=0;
double mean_x=0,mean_y=0;
//计算平均值
for(i=0;i<n;i++)
{
sum1+=x[i];
sum2+=y[i];
}
mean_x=sum1/n;
mean_y=sum2/n;
//计算b
sum1=sum2=0;
for(i=0;i<n;i++)
{
sum1+=(x[i]-mean_x)*(y[i]-mean_y);
sum2+=(x[i]-mean_x)*(x[i]-mean_x);
}
b=sum1/sum2;
//计算a
a=mean_y-b*mean_x;
//输出
printf("拟合的直线方程为y=%f+(%f)*x",a,b);
}
int main()
{
double x[14]={-0.5,-1.0,-1.5,-2.0,-2.5,-3,-3.5,-5,-7,-10,-0.1,-0.2,-0.3,-0.4};
double y[14]={0.528453,1.04233,1.51821,2.03788,2.50965,3.01868,3.48255,4.99680,6.98300