函数说明

polyfit 函数是 numpy 中一个常用一个进行曲线拟合的函数。我用通俗的话说就是，知道了一个一次函数 x 与 y 的值，要求解这个方程其他参数。

今天这个就是讲求解一次函数 y = k * x + b。 就是知道 x 与 y 的一系列值，然后利用 polyfit 函数求解 k 与 b 的值。

参数说明

polyfit 函数常用参数有3个，分别是：x、y、deg。

x：x坐标值的列表；

y：y坐标值的列表；

deg：是方程阶数，也就是自变量的最高次方（y = k * x^deg + b）；

使用说明

我们已知 x 的值是： [1,2,3] 。分别对应的 y 值是：[2,4,6]。这么简单的方程我们可以根据口算得到 y = 2 * x。其中 k = 2，b = 0。

接下来我们使用代码来求解：

import numpy

x = [1,2,3]
y = [2,4,6]
res = numpy.polyfit(x, y, 1)
print(res)

输出结果：

[2.00000000e+00 2.41660629e-15]

第一位就是求解的 k 值，第二位是求解的 b 值。由于计算机中浮点数都不是准确值，所以其中 k 不是绝对的 2，b 也不是绝对的 0，是非常接近 0 的一个值。

阅读原文

了解更多