使用quad方法python进行集成

时间:2016-12-28 17:50:28

标签: python scipy

我试图使用scipy重现错误函数,而不是使用erf方法。

这是我的代码:

#!/usr/bin/env python
import numpy as np
import math
from scipy.integrate import quad

def error_func(y):
    return 2/math.sqrt(np.pi)*quad(np.exp(-y**2), 1, np.inf, args=(y))[0]

g = [error_func(x) for x in np.arange(-1,1,0.2)]

print g

此代码返回以下错误消息:

File "./test.py", line 9, in <module>
    g = [error_func(x) for x in np.arange(-1,1,0.2)]
  File "./test.py", line 7, in error_func
    return 2/math.sqrt(np.pi)*quad(np.exp(-y**2), 1, np.inf, args=(y))[0]
  File "/usr/local/lib/python2.7/site-packages/scipy/integrate/quadpack.py", line 316, in quad
    points)
  File "/usr/local/lib/python2.7/site-packages/scipy/integrate/quadpack.py", line 383, in _quad
    return _quadpack._qagie(func,bound,infbounds,args,full_output,epsabs,epsrel,limit)
quadpack.error: quad: first argument is not callable

如果我理解正确,quad的第一个参数必须是一个函数。我通过它的事情是正确的。 我的代码出了什么问题?

1 个答案:

答案 0 :(得分:1)

你的第一个问题是

np.exp(-y**2)

不是一个功能。这是一个数字;具体来说,它是e ^( - y ^ 2)的值。如果要定义将y映射到np.exp(-y**2)的函数,最简单的方法是使用lambda语法:

lambda y: np.exp(-y**2)

你还有其他问题:

(y)不是args=(y)中的元组;在分组括号中只是y。单元组元组为(y,)

另外,你不应该将args参数传递给quad,因为你的函数除了我们要整合的参数之外不需要任何其他参数。

最后,你的集成界限是错误的;你应该从0集成到错误函数的参数:

def error_func(x):
    return 2 / math.sqrt(np.pi) * quad(lambda y: np.exp(-y**2), 0, x)[0]