正文

python中怎么拟合数据

小六 /31 浏览量
0302

在Python中进行数据拟合，你可以使用`scipy`库中的`curve_fit`函数。下面是一个简单的步骤说明，以及一个使用指数函数拟合数据的示例：

步骤说明

导入库

 import numpy as np from scipy.optimize import curve_fit

定义拟合函数

 def func（x, a, b, c）: return a * np.exp（-b * x） + c

准备数据

 x = np.array（[0, 1, 2, 3, 4, 5]） y = np.array（[1, 2.2, 3.8, 6.5, 10.4, 16.5]）

进行拟合

 params, params_covariance = curve_fit（func, x, y）

打印拟合参数

 print（params）

示例代码

 import numpy as np from scipy.optimize import curve_fit import matplotlib.pyplot as plt 定义拟合函数 def func（x, a, b, c）: return a * np.exp（-b * x） + c 准备数据 x = np.array（[0, 1, 2, 3, 4, 5]） y = np.array（[1, 2.2, 3.8, 6.5, 10.4, 16.5]） 进行拟合 params, params_covariance = curve_fit（func, x, y） 打印拟合参数 print（"拟合参数：", params） 使用拟合参数计算拟合曲线上的值 y_fit = func（x, *params） 绘制原始数据和拟合曲线 plt.scatter（x, y, label='数据'） plt.plot（x, y_fit, label='拟合曲线'） plt.legend（） plt.show（）

以上代码将展示如何使用`curve_fit`函数拟合一组数据，并绘制出拟合曲线。你可以根据自己的数据集和需求修改函数定义和数据准备部分

-- 展开阅读全文 --