使用Numpy获取多项式的系数。

Question

我正在尝试获取通过fit方法获得的numpy.polynomial.polynomial.Polynomial的系数：

import numpy.polynomial as poly

x = [1, 2, 3, 4, 5]
y = [16, 42.25, 81, 132.25, 196]

c = poly.Polynomial.fit(x, y, deg = 2)
print(c(5))
print(c)

这个小程序打印

196.00000000000006
poly([81. 90. 25.])

这是c(5)的正确值，但不是多项式系数的正确值，它们是2.25、7.5和6.25。 我该如何获得实际系数？

- Javier Garcia

Answer 1

根据文档，.fit()方法返回一个表示数据最小二乘拟合的系列，并具有调用中指定的域和窗口。如果有兴趣获取未缩放和未移位基础多项式的系数，请执行new_series.convert().coef操作。在你的数据上运行c.convert().coef会得到：array([2.25, 7.5 , 6.25])。

- AlexK