不同的狡猾的颂歌解说者之间的交换

时间:2015-11-04 13:04:52

标签: python scipy

我有一个可以在scipy.integrate.odescipy.integrate.odeint之间互换的解算器。这是代码。

def f(y,s,C,u,v):
    y0 = y[0] # u
    y1 = y[1] # u'
    y2 = y[2] # v
    y3 = y[3] # v'
    dy = np.zeros_like(y)
    dy[0] = y1
    dy[2] = y3

    C = C.subs({u:y0,v:y2})
    dy[1] = -C[0,0][0]*dy[0]**2\
            -2*C[0,0][1]*dy[0]*dy[2]\
            -C[0,1][1]*dy[2]**2
    dy[3] = -C[1,0][0]*dy[0]**2\
            -2*C[1,0][1]*dy[0]*dy[2]\
            -C[1,1][1]*dy[2]**2
    return dy

def solve(C,u0,s0,s1,ds,solver=None):
    from sympy.abc import u,v
    if solver == None: # use lsoda from scipy.integrate.odeint
        s = np.arange(s0,s1+ds,ds)
        print 'Running solver ...'
        return sc.odeint(f,u0,s,args=(C,u,v))
    else: # use any other solver from scipy.integrate.ode
        r = sc.ode(f).set_integrator(solver) # vode,zvode,lsoda,dopri5,dop853
        r.set_f_params(C,u,v)
        r.set_initial_value(u0)
        #t = []
        y = []
        print 'Running solver ...'
        while r.successful() and r.t <= s1:
            r.integrate(r.t + ds)
            y.append(r.y)#; t.append(r.t)
        return np.array(y)

我遇到的问题如下。如果我决定使用scipy.integrate.odeint中的求解器,那么f的参数必须按照它们在代码中的顺序指定。但是,如果我决定使用scipy.integrate.ode中的解算器,我必须将函数f(y,s,C,u,v)的参数顺序更改为f(s,y,C,u,v),否则我会收到错误

TypeError: 'float' object has no attribute '__getitem__'

如果我执行此操作,则scipy.integrate.odeint会为定义为f的{​​{1}}生成相同的错误。无论参数的顺序如何,我如何使用统一的f(s,y,C,u,v)进行操作?

修改:

总结问题:

如果函数f定义为f,则

scipy.integrate.ode求解器有效,如果函数f定义为f(s,y,C,u,v),则scipy.integrate.odeint求解器有效。为什么会发生这种情况,我该如何解决这个问题?

修改:

Scipy - 版本0.16.0

1 个答案:

答案 0 :(得分:5)

  

为什么会发生这种情况,我该如何解决这个问题?

这是因为多年前做出的令人遗憾的API设计决定。 Cars.Include(x => x.CarModel.StockList) odeint类需要不同的签名才能解决系统问题。

您可以通过添加一个包装来修复它,当您使用ode类时,它会更改前两个参数的顺序。例如,您可以更改此内容:

ode

    r = sc.ode(f).set_integrator(solver)

正在对这种差异进行更好的记录:https://github.com/scipy/scipy/pull/5151