【发布时间】:2020-12-14 13:07:13
【问题描述】:
我花了一些时间,但我使用下面的代码为我的 x,y 数据集创建了一个高斯拟合。
import matplotlib.pyplot as plt
import numpy as np
from scipy.optimize import curve_fit
def Gauss(x, a, x0, sigma, offset):
return a * np.exp(-(x - x0)**2 / (2 * sigma**2)) + offset
x, y = np.random.random(100), np.random.random(100)
popt, pcov = curve_fit(Gauss, x, y, p0=[np.max(y), np.median(x), np.std(x), np.min(y)])
plt.plot(x, y, 'b+:', label='data')
x_fit = np.linspace(np.min(x), np.max(x), 1000)
plt.plot(x_fit, Gauss(x_fit, *popt), 'r-', label='fit')
plt.legend()
plt.title('Something')
plt.xlabel('Anotherthing')
plt.ylabel('Athing')
plt.show()
我可以看到我的合身状态很好,并且可以查看图表和所有内容。
我现在想知道的是如何在我的屏幕上打印出这种拟合的结果,例如拟合最大值点 x 处的最大值、估计误差等?
这些信息可以访问吗?如果是这样,有没有办法打印出这些信息?如果没有,有人可以指出我找到合适的错误的正确方向吗?
【问题讨论】:
-
你能添加你看到的图像吗?甚至更好 - 让它成为一个可重复的例子?
-
@VladimírKunc 我已经添加了这个数字。我调用的数据文件只包含 x 和 y 数字。
-
@VladimírKunc 您只需将第 6 行替换为 x, y = np.random.random(100), np.random.random(100) 即可获得工作示例代码。跨度>
标签: python scipy curve-fitting data-fitting gauss