高斯函数拟合是一种非线性回归方法,用于拟合数据分布,在Python中,我们可以使用scipy.optimize.curve_fit
函数进行高斯函数拟合,以下是详细的步骤和小标题:
1、导入所需库
2、定义高斯函数
3、准备数据
4、使用curve_fit进行拟合
5、可视化结果
1. 导入所需库
import numpy as np import matplotlib.pyplot as plt from scipy.optimize import curve_fit
2. 定义高斯函数
def gaussian(x, a, x0, sigma): return a * np.exp((x x0)**2 / (2 * sigma**2))
3. 准备数据
生成模拟数据 x_data = np.linspace(10, 10, 100) y_data = gaussian(x_data, 1, 0, 1) + np.random.normal(0, 0.1, len(x_data))
4. 使用curve_fit进行拟合
popt, pcov = curve_fit(gaussian, x_data, y_data)
popt
是一个包含拟合参数的数组,pcov
是协方差矩阵。
5. 可视化结果
绘制原始数据和拟合曲线 plt.scatter(x_data, y_data, label='Data') plt.plot(x_data, gaussian(x_data, *popt), 'r', label='Fit: a=%5.3f, x0=%5.3f, sigma=%5.3f' % tuple(popt)) plt.legend() plt.show()
通过以上步骤,我们可以实现高斯函数拟合。
最新评论
本站CDN与莫名CDN同款、亚太CDN、速度还不错,值得推荐。
感谢推荐我们公司产品、有什么活动会第一时间公布!
我在用这类站群服务器、还可以. 用很多年了。