多项式曲线拟合，c代码，可实现1阶线性，2-4阶多项式曲线拟合，代码注释详细，方便移植，书写规范 图片有现场拟合参数的1-4阶的keil仿真结果和Excel对照图。 备注一下，这是个多项式求解代码，求每个相的系数
在数据处理和分析中，多项式曲线拟合是一种非常有用的技术。它可以帮助我们找到一条曲线，尽可能地贴近给定的数据点，从而对数据进行建模和预测。今天咱们就聊聊如何用C代码实现1阶线性以及2 - 4阶多项式曲线拟合。
多项式曲线拟合原理简述
简单来说，对于给定的一组数据点 $(xi, yi)$，$i = 1, 2, \cdots, n$，我们想要找到一个多项式函数：
$y = a0 + a1x + a2x^2 + \cdots + amx^m$
使得这个多项式函数在这些数据点上的误差尽可能小。通常使用最小二乘法来确定多项式的系数 $a0, a1, \cdots, a_m$。最小二乘法的核心思想是让所有数据点到拟合曲线的误差的平方和最小。
C代码实现
#include <stdio.h> #include <stdlib.h> #include <math.h> // 函数声明 void polynomial_fit(double *x, double *y, int n, int degree, double *coeff); int main() { // 示例数据点 double x[] = {1.0, 2.0, 3.0, 4.0, 5.0}; double y[] = {2.0, 4.0, 6.0, 8.0, 10.0}; int n = sizeof(x) / sizeof(x[0]); int degree = 1; // 1阶线性拟合 double *coeff = (double *)malloc((degree + 1) * sizeof(double)); polynomial_fit(x, y, n, degree, coeff); printf("拟合多项式系数: "); for (int i = 0; i <= degree; i++) { printf("%lf ", coeff[i]); } printf("\n"); free(coeff); return 0; } // 多项式拟合函数 void polynomial_fit(double *x, double *y, int n, int degree, double *coeff) { // 初始化系数矩阵和结果向量 double **a = (double **)malloc((degree + 1) * sizeof(double *)); for (int i = 0; i <= degree; i++) { a[i] = (double *)malloc((degree + 1) * sizeof(double)); } double *b = (double *)malloc((degree + 1) * sizeof(double)); // 填充系数矩阵a和结果向量b for (int i = 0; i <= degree; i++) { for (int j = 0; j <= degree; j++) { a[i][j] = 0.0; for (int k = 0; k < n; k++) { a[i][j] += pow(x[k], i + j); } } b[i] = 0.0; for (int k = 0; k < n; k++) { b[i] += pow(x[k], i) * y[k]; } } // 使用高斯消元法求解线性方程组 for (int k = 0; k <= degree; k++) { double max = fabs(a[k][k]); int max_row = k; for (int i = k + 1; i <= degree; i++) { if (fabs(a[i][k]) > max) { max = fabs(a[i][k]); max_row = i; } } if (max_row!= k) { for (int j = 0; j <= degree; j++) { double temp = a[k][j]; a[k][j] = a[max_row][j]; a[max_row][j] = temp; } double temp = b[k]; b[k] = b[max_row]; b[max_row] = temp; } for (int i = k + 1; i <= degree; i++) { double factor = a[i][k] / a[k][k]; for (int j = 0; j <= degree; j++) { a[i][j] -= factor * a[k][j]; } b[i] -= factor * b[k]; } } // 回代求解系数 for (int i = degree; i >= 0; i--) { coeff[i] = b[i]; for (int j = i + 1; j <= degree; j++) { coeff[i] -= a[i][j] * coeff[j]; } coeff[i] /= a[i][i]; } // 释放内存 for (int i = 0; i <= degree; i++) { free(a[i]); } free(a); free(b); }代码分析
- 头文件包含：
-stdio.h用于标准输入输出，比如printf函数。
-stdlib.h提供了动态内存分配函数malloc和free，以及程序控制函数exit。
-math.h包含了数学运算函数，像pow用于计算幂次方。
- 函数声明：
polynomial_fit函数用于执行多项式拟合，在main函数调用前先声明。
main函数：
- 定义了示例数据点x和y，以及数据点的数量n。
- 设置拟合的阶数degree，这里先设为1阶线性拟合。
- 使用malloc为拟合系数分配内存空间。
- 调用polynomial_fit函数进行拟合，并输出拟合得到的多项式系数。
- 最后使用free释放分配的内存，避免内存泄漏。
polynomial_fit函数：
- 初始化系数矩阵a和结果向量b，并使用malloc分配内存。
- 通过循环填充系数矩阵a和结果向量b的值，这里用到了最小二乘法的数学原理。
- 使用高斯消元法求解线性方程组，过程中包括寻找主元、交换行以及消元操作。
- 通过回代的方式求解出多项式的系数coeff。
- 最后释放之前分配的内存。
Keil仿真结果与Excel对照图
在实际应用中，我们可以通过Keil等工具对代码进行仿真运行。从仿真结果中可以得到不同阶数多项式拟合的系数值。同时，将同样的数据导入Excel，利用Excel的曲线拟合功能，可以得到对应的拟合曲线和系数。对比两者的结果，可以验证我们代码实现的正确性。例如1阶线性拟合时，Keil仿真得到的系数和Excel拟合得到的系数在精度范围内应该是一致的，2 - 4阶多项式拟合同样如此。这样的对比不仅能验证代码，还能加深对多项式曲线拟合原理的理解。
通过上述C代码实现和相关分析，相信大家对多项式曲线拟合有了更深入的认识，并且可以根据实际需求方便地移植这段代码到自己的项目中。
