使用Python做科学计算初探

今天在搞定Django框架的blog搭建后，尝试一下python的科学计算能力。

python的科学计算有三剑客：numpy，scipy，matplotlib。

numpy负责数值计算，矩阵操作等；

scipy负责常见的数学算法，插值、拟合等；

matplotlib负责画图。

 

首先，百度上头三个，依次安装。

可以考虑使用pyhton34/script/easy-install 工具； easy-insatll -m matplotlib;

 

尝试一下代码，拟合实例；

 

 1  #  -*- coding: utf-8 -*-
 2  import numpy as np
 3  from scipy.optimize  import leastsq
 4  import pylab as pl
 5 
 6  def func(x, p):
 7      #  """
 8       #  数据拟合所用的函数: A*sin(2*pi*k*x + theta)
 9       #  """
10      A, k, theta = p
11      return A*np.sin(2*np.pi*k*x+theta)   
12 
13  def residuals(p, y, x):
14      #  """
15       #  实验数据x, y和拟合函数之间的差，p为拟合需要找到的系数
16       #  """
17       return y - func(x, p)
18 
19 x = np.linspace(0, -2*np.pi, 100)
20 A, k, theta = 10, 0.34, np.pi/6  #  真实数据的函数参数
21  y0 = func(x, [A, k, theta])  #  真实数据
22  y1 = y0 + 2 * np.random.randn(len(x))  #  加入噪声之后的实验数据    
23 
24 p0 = [7, 0.2, 0]  #  第一次猜测的函数拟合参数
25 
26  #  调用leastsq进行数据拟合
27  #  residuals为计算误差的函数
28  #  p0为拟合参数的初始值
29  #  args为需要拟合的实验数据
30  plsq = leastsq(residuals, p0, args=(y1, x))
31 
32  #  print (u"真实参数:")
33  print([A, k, theta])
34  #  print (u"拟合参数")
35  print(plsq[0])  #  实验数据拟合后的参数
36 
37 pl.plot(x, y0, label=u " real data ")
38 pl.plot(x, y1, label=u " data with noisy ")
39 pl.plot(x, func(x, plsq[0]), label=u " nihe data ")
40 pl.legend()
41 pl.show()

运行提示错误，缺少第三方包，如six,dateutil,pyparsing等，缺什么装什么；第三方包大多直接拖到D:\python34\lib目录下就可以了，很方便。

都装上后，运行成功，如图；

 

转载于:https://www.cnblogs.com/javajava/p/4792791.html