如何在python3中平滑线条?

问题描述 投票:1回答:1

请如何平滑这条蓝线?

import numpy as np
from matplotlib import pyplot as plt

a = [0.5365140382771445, 0.5214107372135204, 0.49844631119258076, 0.4681910992517213, 0.4310817214420628, 0.3882500155177606, 0.340292343154754, 0.2880252732908801]
b = [0.7416012836460293, 0.697385422521102, 0.6561831711375956, 0.6187959941327967, 0.585900754784896, 0.5586375446776617, 0.537388969490203, 0.5229339200070606]

time_fit = np.arange(len(b))
fitx = np.polyfit(time_fit, a, 2)
fity = np.polyfit(time_fit, b, 2)
x_fit = np.poly1d(fitx)
y_fit = np.poly1d(fity)

plt.figure(figsize=(8, 8))
plt.plot(x_fit, y_fit, c='red')
plt.plot(a, b, c = 'blue')
plt.show()

我是否使用错误的多项式拟合?

enter image description here

python-3.x matplotlib smoothing
1个回答
0
投票

因此,您正在尝试将多项式曲线拟合到您的点。让我们分解一下二维点的曲线拟合的工作原理。为简单起见,我选取了二阶多项式来拟合这些点。另外,我还可以自由命名点名称(a-> x和b-> y)。

np.polyfit返回指定顺序的多项式系数。您可以通过将np.poly1d应用于返回的系数来制作实际的多项式类。为了绘制多项式,我们拟合x并得到z(x,z)点是我们拟合的曲线。

import numpy as np
from matplotlib import pyplot as plt


x = [
    0.5365140382771445,
    0.5214107372135204,
    0.49844631119258076,
    0.4681910992517213,
    0.4310817214420628,
    0.3882500155177606,
    0.340292343154754,
    0.2880252732908801,
]
y = [
    0.7416012836460293,
    0.697385422521102,
    0.6561831711375956,
    0.6187959941327967,
    0.585900754784896,
    0.5586375446776617,
    0.537388969490203,
    0.5229339200070606,
]

# fitting x, y in a second order equation
# this returns the coefficient of p_0*x^2 + p_1*x + p_n = 0
# building the polynomial
poly = np.poly1d(np.polyfit(x, y, 2))

# getting the points for plotting polynomial
z = poly(x)

plt.figure(figsize=(8, 8))

# scatter plotting the actual value
plt.scatter(x, y, c="blue", marker="o", label="original points")

# plotting the polynomial curve
plt.plot(x, z, c="red", label="fitted curve (2nd order)")
plt.legend()
plt.show()

此返回,

enter image description here

© www.soinside.com 2019 - 2024. All rights reserved.