直線擬合算法

函數說明：

通過一些點擬合出一條直線。

參數：

pt_input
指向傳入的點的指針。

ptNumbers
傳入的點數量。

k
指向擬合直線參數k的指針。

b
指向擬合直線參數b的指針。

返回值：

返回一個代數值判斷擬合是否成功。如果成功，則返回 0。如果返回-1，表示點的數量<2。如果返回-2，表示x坐標無變化。

代碼：

typedef struct tagPOINT_2 {double x;double y; }POINT_2, *PPOINT_2;int LineInfo(POINT_2 *pt_input, int ptNumbers, double * k, double * b) {int nRet = 0;double X_aver = 0;double Y_aver = 0;double A = 0;double B = 0;double _2_XY_av = 0;double _2_XX_av = 0;do{if (ptNumbers < 2){nRet = -1;break;}for (int loop = 0; loop < ptNumbers; loop++){X_aver += pt_input[loop].x;Y_aver += pt_input[loop].y;_2_XY_av += pt_input[loop].x * pt_input[loop].y;_2_XX_av += pt_input[loop].x * pt_input[loop].x;}X_aver /= ptNumbers;Y_aver /= ptNumbers;_2_XY_av /= ptNumbers;_2_XX_av /= ptNumbers;for (int loop = 0; loop < ptNumbers; loop++){A += fabs((pt_input[loop].x - X_aver) * (pt_input[loop].y - Y_aver));B += fabs((pt_input[loop].x - X_aver) * (pt_input[loop].x - X_aver));}if (B < 0.000001){nRet = -2;break;}//方式一//*k = A / B;//*b = Y_aver - *k * X_aver;//方式二*k = (_2_XY_av - X_aver * Y_aver) / (_2_XX_av - X_aver * X_aver);*b = Y_aver - (*k) * X_aver;} while (0);return nRet; }

測試代碼：

#include <stdio.h> #include <math.h> int main() {POINT_2 pt[10] = {{0.1 , 0.3},{1.2 , 2.0},{2.2 , 4.3},{3.1 , 5.8},{4.0 , 8.5},{5.3 , 10.8},{5.9 , 11.1},{6.8 , 13.9},{8.1 , 16.5},{9.0 , 17.9},};int ret;double k = 0, b = 0;ret = LineInfo(pt, 10, &k, &b);switch (ret){case 0: printf("擬合成功，直線為y = %lf * x + %lf\n", k, b); break;case -1: printf("擬合失敗，傳入的點少于二點\n"); break;case -2: printf("擬合失敗，傳入的點x坐標重合\n"); break;default:printf("error\n"); break;}return 0; }

測試結果：

總結

以上是生活随笔為你收集整理的C语言直线拟合函数的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。