python - numpy.polyfit 的错误是什么？

Question

我想使用 numpy.polyfit 进行物理计算，因此我需要误差的大小。

Answer 1

如果您在调用 polyfit 时指定 full=True ，它将包含额外的信息:

>>> x = np.arange(100)
>>> y = x**2 + 3*x + 5 + np.random.rand(100)
>>> np.polyfit(x, y, 2)
array([ 0.99995888,  3.00221219,  5.56776641])
>>> np.polyfit(x, y, 2, full=True)
(array([ 0.99995888,  3.00221219,  5.56776641]), # coefficients
 array([ 7.19260721]), # residuals
 3, # rank
 array([ 11.87708199,   3.5299267 ,   0.52876389]), # singular values
 2.2204460492503131e-14) # conditioning threshold

返回的残差是拟合误差的平方和，不知道是不是你要的:

>>> np.sum((np.polyval(np.polyfit(x, y, 2), x) - y)**2)
7.1926072073491056

在 1.7 版中还有一个 cov 关键字，它将返回系数的协方差矩阵，您可以使用它来计算拟合系数本身的不确定性。