在没有numpy polyfit的python中拟合二次函数

Question

我正在尝试将二次函数拟合到某些数据中，并且我正在尝试在不使用 numpy 的 polyfit 函数的情况下做到这一点。

从数学上讲，我尝试关注这个网站https://neutrium.net/mathematics/least-squares-fitting-of-a-polynomial/，但不知何故，我认为我做得不对。如果有人可以帮助我，那就太好了，或者如果您能提出另一种方法，那也很棒。

到目前为止我已经尝试过：

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

ones = np.ones(3)
A = np.array( ((0,1),(1,1),(2,1)))
xfeature = A.T[0]
squaredfeature = A.T[0] ** 2
b = np.array( (1,2,0), ndmin=2 ).T
b = b.reshape(3)

features = np.concatenate((np.vstack(ones), np.vstack(xfeature), np.vstack(squaredfeature)), axis = 1)
featuresc = features.copy()
print(features)
m_det = np.linalg.det(features)
print(m_det)
determinants = []
for i in range(3):
    featuresc.T[i] = b
    print(featuresc)
    det = np.linalg.det(featuresc)
    determinants.append(det)
    print(det)
    featuresc = features.copy()

determinants = determinants / m_det
print(determinants)
plt.scatter(A.T[0],b)
u = np.linspace(0,3,100)
plt.plot(u, u**2*determinants[2] + u*determinants[1] + determinants[0] )
p2 = np.polyfit(A.T[0],b,2)
plt.plot(u, np.polyval(p2,u), 'b--')
plt.show()

如您所见，我的曲线与 nnumpy 的 polyfit 曲线相比不太好。

---

更新： 我检查了我的代码并删除了所有愚蠢的错误，现在它可以工作了，当我尝试将其拟合超过 3 分时，但我不知道如何拟合超过 3 分。

这是新代码：

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

ones = np.ones(3)
A = np.array( ((0,1),(1,1),(2,1)))
xfeature = A.T[0]
squaredfeature = A.T[0] ** 2
b = np.array( (1,2,0), ndmin=2 ).T
b = b.reshape(3)

features = np.concatenate((np.vstack(ones), np.vstack(xfeature), np.vstack(squaredfeature)), axis = 1)
featuresc = features.copy()
print(features)
m_det = np.linalg.det(features)
print(m_det)
determinants = []
for i in range(3):
    featuresc.T[i] = b
    print(featuresc)
    det = np.linalg.det(featuresc)
    determinants.append(det)
    print(det)
    featuresc = features.copy()

determinants = determinants / m_det
print(determinants)
plt.scatter(A.T[0],b)
u = np.linspace(0,3,100)
plt.plot(u, u**2*determinants[2] + u*determinants[1] + determinants[0] )
p2 = np.polyfit(A.T[0],b,2)
plt.plot(u, np.polyval(p2,u), 'r--')
plt.show()

Answer 1

而不是使用克莱默规则，实际上使用最小二乘法求解系统。请记住，只有当您拥有的总点数等于所需的多项式阶数加 1 时，克莱默规则才有效。 如果你没有这个，那么在你试图找到问题的确切解决方案时，克莱默规则将不起作用。如果您有更多点，则该方法不合适，因为我们将创建一个超定方程组。

为了适应更多的点，numpy.linalg.lstsq 会更合适，因为它通过计算向量 x 来解决 Ax = b使用矩阵 A 的欧几里得范数。因此，从特征矩阵的最后一列中删除y的值并求解系数，并使用numpy.linalg.lstsq求解系数：

import numpy as np
import matplotlib.pyplot as plt


ones = np.ones(4)
xfeature = np.asarray([0,1,2,3])
squaredfeature = xfeature ** 2
b = np.asarray([1,2,0,3])

features = np.concatenate((np.vstack(ones),np.vstack(xfeature),np.vstack(squaredfeature)), axis = 1) # Change - remove the y values

determinants = np.linalg.lstsq(features, b)[0] # Change - use least squares
plt.scatter(xfeature,b)
u = np.linspace(0,3,100)
plt.plot(u, u**2*determinants[2] + u*determinants[1] + determinants[0] )
plt.show()

我现在得到了这个图，它与您的图表中的虚线相匹配，也与numpy.polyfit 给您的相匹配：