从一些数据生成二次多项式

Question

我有一些数据想要生成像这样的二度polyfit：

我尝试了两种不同的代码，但多项式只是试图遍历所有点。

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

data = pd.read_csv('TESTEXskelet.csv', sep=",")

x = data.Gennemsnitlig_hastighed
y1 = data.Sum_VSP

np.polyfit(x,y1,2)

plt.grid()
plt.title("VSP sum/hastighed")
plt.ylabel('VSP - kW/ton')
plt.xlabel('Hastighed - km/t')

plt.scatter(x,y1,s=5) # Definere selve plottet
plt.plot(x, y1)

但随后它会通过每个点来绘制它。

我也尝试过使用 sklearn，如果需要的话我可以上传。

Answer 1

您正确拟合了二次多项式。你只是在那之后的情节中没有使用它。

plt.scatter(x,y1,s=5) 绘制原始数据的散点图，plt.plot(x, y1) 绘制一条穿过所有数据的线数据。

要绘制多项式，您需要将多项式拟合到变量中。然后定义要绘制的 x 轴范围并根据多项式拟合预测 y 值：

p = np.polyfit(x,y1,2)

xn = np.linspace(np.min(x), np.max(x), 100)
yn = np.poly1d(p)(xn)

plt.scatter(x,y1,s=5)
plt.plot(xn, yn)

Answer 2

polyfit 将参数返回到您的多项式，尝试

p = np.polyfit(x,y1,2)
y2 = np.polyval(p, x)
plt.plot(x, y2)