在 Python 中获取多项式对数据的最小二乘拟合

要获取多项式对数据的最小二乘拟合，请在 Python Numpy 中使用 polynomial.polyfit()。该方法返回从低到高排序的多项式系数。如果 y 是二维的，则 coef 中第 k 列的系数表示对 y 的第 k 列中的数据进行的多项式拟合。参数 x 是 M 个样本（数据）点 (x[i], y[i]) 的 x 坐标。

参数 y 是样本点的 y 坐标。通过为 y 传递包含每个数据集中一列的二维数组，可以（独立地）在一个对 polyfit 的调用中拟合共享相同 x 坐标的多个样本点集。参数 deg 是拟合多项式的次数。如果 deg 是一个单个整数，则拟合中将包含直至并包括 deg 次项的所有项。

参数 rcond 是拟合的相对条件数。相对于最大奇异值，小于 rcond 的奇异值将被忽略。默认值为 len(x)*eps，其中 eps 是平台浮点类型的相对精度，在大多数情况下约为 2e-16。参数 full 是确定返回值性质的开关。当为 False（默认值）时，仅返回系数；当为 True 时，还将返回来自奇异值分解的诊断信息。

参数 w 是权重。如果非 None，则权重 w[i] 应用于 x[i] 处未平方残差 y[i] - y_hat[i]。理想情况下，权重的选择应使 w[i]*y[i] 的乘积误差都具有相同的方差。当使用逆方差加权时，使用 w[i] = 1/sigma(y[i])。默认值为 None。

步骤

首先，导入所需的库 -

import numpy as np
from numpy.polynomial import polynomial as P

x 坐标 -

x = np.linspace(-1,1,51)

显示 x 坐标 -

print("X Co-ordinate...\n",x)

y 坐标 -

y = x**3 - x + np.random.randn(len(x))
print("\nY Co-ordinate...\n",y)

要获取多项式对数据的最小二乘拟合，请在 Python Numpy 中使用 polynomial.polyfit() -

c, stats = P.polyfit(x,y,3,full=True)
print("\nResult...\n",c)
print("\nResult...\n",stats)

示例

import numpy as np
from numpy.polynomial import polynomial as P

# The x-coordinate
x = np.linspace(-1,1,51)

# Display the x-coordinate
print("X Co-ordinate...\n",x)

# The y-coordinate
y = x**3 - x + np.random.randn(len(x))
print("\nY Co-ordinate...\n",y)

# To get the least-squares fit of a polynomial to data, use the polynomial.polyfit() in Python Numpy

# The method returns the Polynomial coefficients ordered from low to high. If y was 2-D, the coefficients in column k of coef represent the polynomial fit to the data in y’s k-th column.
c, stats = P.polyfit(x,y,3,full=True)
print("\nResult...\n",c)
print("\nResult...\n",stats)

输出

X Co-ordinate...
[-1. -0.96 -0.92 -0.88 -0.84 -0.8 -0.76 -0.72 -0.68 -0.64 -0.6 -0.56
 -0.52 -0.48 -0.44 -0.4 -0.36 -0.32 -0.28 -0.24 -0.2 -0.16 -0.12 -0.08
 -0.04 0. 0.04 0.08 0.12 0.16 0.2 0.24 0.28 0.32 0.36 0.4
  0.44 0.48 0.52 0.56 0.6 0.64 0.68 0.72 0.76 0.8 0.84 0.88
  0.92 0.96 1. ]

Y Co-ordinate...
[ 0.24282445 -0.48320773 -1.18068109 -0.48486683 0.80514762 0.48976259
  0.31363813 -0.03382685 -0.92334611 0.86835062 0.24661201 0.9790978
  0.03782101 0.12213365 -1.37248029 1.99891304 -0.09825977 1.74524931
  0.70560858 0.15516069 0.69169705 0.76957712 -1.21919676 1.50064825
  1.32101339 -2.51479992 -0.28998783 -1.24293076 0.45927699 -0.53484746
  0.50455341 -0.06351788 -2.69585303 -0.46833578 1.4924168 -2.42374146
 -1.91934499 -1.36311466 -1.23946547 -1.56822005 -0.79648036 1.58269324
 -0.53682862 -0.90861958 -0.28174461 -0.10775622 0.58535687 1.06373501
 -2.28991738 2.01597286 -0.75841069]

Result...
[-0.17198829 -1.84107674 0.09439374 2.39030912]

Result...
[array([60.43653521]), 4, array([1.38446749, 1.32119158, 0.50443316, 0.28853036]), 1.1324274851176597e-14]

更新于: 2022-02-28

8K+ 次查看

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告