Python数据拟合实战:scipy.optimize.curve_fit详解

陈默
陈默
发布时间:2026-10-16 | 7人浏览过
原创

本文深入解析Python中scipy.optimize.curve_fit函数的用法,对比其与numpy polyfit在拟合范围上的差异。通过二次多项式拟合示例,展示如何定义目标函数、获取拟合参数a,b,c及协方差矩阵,并演示利用拟合结果进行数据预测与可视化绘图,帮助开发者掌握更灵活的数据建模技巧。

scipy.optimize.curve_fit解读

相较于numpy库中仅支持多项式拟合的polyfit函数,scipy.optimize模块提供的curve_fit函数具备更广泛的适用性,能够处理非多项式类型的复杂函数拟合任务。

已有数据

# 导入模块
import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit
# 导入数据
x = np.linspace(0,1.0,11)
y = np.array([-0.447, 1.978, 3.28, 6.16, 7.08, 7.34, 7.66, 9.56, 9.48, 9.30, 11.2])
# 拟合二次多项式
f = lambda x, a, b, c: a*x**2 + b*x + c
 
# popt为拟合参数,pcov为参数的协方差矩阵
popt, pcov = curve_fit(f, x, y)
print("拟合的参数值a,b,c分别为:", popt)
 
# x=0.25, 0.35的预测值
print("x=0.25, 0.35的预测值为", f(np.array([0.25, 0.35]), *popt))

拟合得到的参数值a、b、c分别为:[-9.81083901 20.12929291 -0.03167108]

当x取值为0.25和0.35时,对应的预测结果分别为[4.38747471 5.81175366]

plt.rc('font', size=15)
plt.plot(x, y, '*')    # 观测值数据的散点图
# 拟合数据的折线图
plt.plot(np.linspace(0.0, 1.0, 50), f(np.linspace(0.0, 1.0, 50), *popt), '-')
plt.show()

总结

以上为个人经验总结,旨在为读者提供参考,欢迎大家持续关注本站获取更多内容。

相关标签:

本站声明

本文内容仅供参考,不代表本站观点。转载或引用请注明出处。如发现内容问题,请通过联系我们页面反馈。

Python 相关产品和应用

全部下载

专题列表