Python实现曲线拟合操作示例【基于numpy,scipy,matplotlib库】

本文实例讲述了Python实现曲线拟合操作。分享给大家供大家参考，具体如下：

这两天学习了用python来拟合曲线。

一、环境配置

本人比较比较懒，所以下载的全部是exe文件来安装，安装按照顺利来安装。自动会找到python的安装路径，一直点下一步就行。还有其他的两种安装方式：一种是解压，一种是pip。我没有尝试，就不乱说八道了。

没有ArcGIS 环境的，可以不看下面这段话了。

在配置环境时遇见一个小波折，就是原先电脑装过ArcGIS10.2 ,所以其会默认安装python2.7，而且python是32位的。且其目录为C:/Python27/ArcGIS10.2，所以引用环境变量时，要注意。并且在其引用的工具包中本身包含numpy，matplotlib的包。还是很方便的。但是因为之前想用PyQT来做曲线拟合的界面，安装QT时总是失败，所以最后放弃使用这个。在安装新的python时注意要把路径写到上面这个路径前面，意思就是说在安装上面的包的时候会找默认python路径。我新安装python路径为C:/Python2，要不然就会找到ArcGIS那个python包路径下了。

1. 安装包

python2.7 (32位)https://www.python.org/downloads/
numpy-1.8.1-win32-superpack-python2.7
scipy-0.15.1-win32-superpack-python2.7
matplotlib-1.3.1.win32-py2.7

安装过程中遇见的问题

提示

numpy是Python的一种开源的数值计算扩展，数学计算很方便。
scipy是一款方便、易于使用、专为科学和工程设计的Python工具包.它包括统计,优化,整合,线性代数模块,傅里叶变换,信号和图像处理,常微分方程求解器等等.这次还没用到，看介绍很强大，是numpy的升级版。

matplotlib是Python的一种开源的扩展可以绘制各种各种的图表。

二、实例

曲线拟合的例子

import matplotlib.pyplot as pltimport mathimport numpy as npimport randomimport csvplt.rcParams['font.sans-serif'] = ['SimHei']#设置显示中文fig = plt.figure()ax = fig.add_subplot(111)#将画布分割成1行1列，图像画在从左到右从上到下的第1块#阶数为6阶order=6#生成曲线上的各个点dataMat = np.loadtxt(open("c://yandu.csv","rb"),delimiter=",",skiprows=0)size=dataMat.shapenum=size[0]trandata=np.transpose(dataMat)#矩阵转置xa=trandata[0]#得到天数数组（横坐标）ya=trandata[1]#实测盐度值数组#数据筛选,去除盐度值为零的，提高拟合精度i=0x=[]y=[]for yy in ya:  if yy>0:    xx=xa[i]    i+=1    x.append(xx)    y.append(yy)#绘制原始数据ax.plot(x,y,label=u'原始数据',color='m',linestyle='',marker='.')#计算多项式c=np.polyfit(x,y,order)#拟合多项式的系数存储在数组c中yy=np.polyval(c,x)#根据多项式求函数值#进行曲线绘制x_new=np.linspace(0, 365, 2000)f_liner=np.polyval(c,x_new)#ax.plot(x,y,color='m',linestyle='',marker='.')ax.plot(x_new,f_liner,label=u'拟合多项式曲线',color='g',linestyle='-',marker='')# labels标签设置ax.set_xlim(0, 366)ax.set_xlabel(u'天')ax.set_ylabel(u'盐度')ax.set_title(u'盐度的日变化', bbox={'facecolor':'0.8', 'pad':5})ax.legend()plt.show()