2016-11-17 162 views
1

我正试图对插值函数进行双重集成,其中r = r(x,y)python集成内插函数(interp1d)

from scipy import interpolate 
import scipy as sp 

r = [0, 1, 2] 
z = [0, 1, 2] 

def cartesian(x, y, f): 
    r = sp.sqrt(x**2 + y**2) 
    return f(r) 

interp = interpolate.interp1d(r, z) 

print(cart(1,1,interp)) 

a = sp.integrate.dblquad(cart, 0, 1, lambda x: 0, lambda x: 1, args=(interp)) 
print(a) 

执行笛卡尔函数一次会产生正确的答案。但是整体给人的以下错误:

TypeError: integrate() argument after * must be an iterable, not interp1d 

我不明白为什么我的功能不是iterable,不知道如何将它转换成一个iterable形式。非常感谢您的帮助。

回答

1

args应该是一个序列参数,所以:

sp.integrate.dblquad(cart, 0, 1, lambda x: 0, lambda x: 1, args=(interp,)) 

interp后的逗号是关键:在Python,(x)只是x,但(x,)是一个元组(即序列) 。