一元N次多项式曲线拟合(求多项式参数)java实现

近期一客户提出需求,要在我们软件中实现一元二次函数的曲线拟合。这个内容之前没有接触过,有点摸不着边。

需求具体一点就是:给出一组数据(x1,y1), (x2,y2), (x3,y3)....(xn,yn),求多项式 y = ax² + bx +c 中的a、b、c三个参数分别是多少,java实现。

严格来说,a, b, c三个参数只是根据现有数据和特定公式拟合出来的,属于“估计”值。

首先从百度查,基本查不到;

然后用bing,英文查,大概找到了一点门道,了解到了Apache的commons-math这个包;

然后再用百度查commons-math,然后就找到了不少内容, 比如下面这篇文章:

https://www.jb51.net/article/126265.htm

经过一番尝试,实际发现可以非常简单地实现(查到的文章和博客写的代码都太多了,而且雷同),代码如下:

一元N次多项式曲线拟合(求多项式参数)java实现

import org.apache.commons.math3.fitting.PolynomialCurveFitter;
import org.apache.commons.math3.fitting.WeightedObservedPoints;

//...

double[] x = new double[]{1, 2, 3, 4, 5};
double[] y = new double[]{19,33,53,79,111};
		  
WeightedObservedPoints points = new WeightedObservedPoints(); 	 
for(int i = 0; i < x.length; i++) { //把数据点加入观察的序列
	points.add(x[i], y[i]);
}
		
PolynomialCurveFitter fitter = PolynomialCurveFitter.create(2);  //指定多项式阶数 
double[] result = fitter.fit(points.toList());  // 曲线拟合,结果保存于数组
		 
for(int i = 0; i < result.length; i++) {
	System.out.println(result[i]);
}

//本例输出:
/*
10.999999999999996
 5.000000000000003
 2.9999999999999996
*/

用的测试公式为 y = 3x² + 5x + 11, 数据点入上面代码中所示,拟合出来的三个参数根据阶数由低到高为:

2.999,5.000,10.999 

基本正确。

这次收获不小。

上一篇:POI操作报错-- java.lang.NoClassDefFoundError: org/apache/commons/math3/util/ArithmeticUtils 解决


下一篇:SpringMVC实现文件上传