python - np.polyfit 做什么并返回？

Question

我浏览了文档，但无法正确解释

在我的代码中，我想找到一条通过 2 个点 (x1,y1), (x2,y2) 的直线，所以我使用了 np.polyfit((x1,x2),(y1,y2),1)因为它是一个 1 度多项式(一条直线)

它返回我 [ -1.04 727.2 ]虽然我的代码(这是一个更大的文件)运行正常，并且按照预期的方式运行 - 我想了解它返回的是什么

我假设 polyfit 返回满足(通过)给它的点的线(弯曲的、直线的，等等)，那么如何用它返回的 2 个点来表示一条线？

Answer 1

来自numpy.polyfit文档:

Returns:

p : ndarray, shape (deg + 1,) or (deg + 1, K)

Polynomial coefficients, highest power first. If y was 2-D, the coefficients for k-th data set are in p[:,k].

所以这些数字是多项式的系数。因此，在您的情况下:

y = -1.04*x + 727.2

顺便说一句，如果多项式的次数至少为 N-1，则 numpy.polyfit 只会返回一个通过所有点(假设您有 N 个点)的方程。否则，它将返回最小化平方误差的最佳拟合。