如何使用“ scipy.optimize.curve_fit”使数据点平滑拟合？

Question

我想使用scipy.optimize.curve_fit来拟合一些数据点。不幸的是，我身体不稳定，我也不知道为什么。

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

M = np.array([730,910,1066,1088,1150], dtype=float)
V = np.array([95.71581923, 146.18564513, 164.46723727, 288.49796413, 370.98703941], dtype=float)

def func(x, a, b, c):
    return a * np.exp(b * x) + c

popt, pcov = curve_fit(func, M, V, [0,0,1], maxfev=100000000)
print(*popt)

fig, ax = plt.subplots()
fig.dpi = 80

ax.plot(M, V, 'go', label='data')
ax.plot(M, func(M, *popt), '-', label='fit')

plt.xlabel("M")
plt.ylabel("V")
plt.grid()
plt.legend()
plt.show()

我非常期待某种平滑的曲线。有人可以在这里解释我在做什么错吗？

Answer 1

您仅在通话中绘制与原始数据相同的x点：

ax.plot(M, V, 'go', label='data')
ax.plot(M, func(M, *popt), '-', label='fit')

要解决此问题，您可以使用更广泛的范围-在这里，我们使用700到1200之间的所有值：

toplot = np.arange(700,1200)
ax.plot(toplot, func(toplot, *popt), '-', label='fit')