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

Question

我有一个可以在scipy.integrate.ode和之间互换的解算器scipy.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定义为f(s,y,C,u,v).f无论参数的顺序如何,我如何操作统一？

编辑:

总结一下这个问题:

scipy.integrate.ode如果函数f被定义为求解器,则求解器起作用,如果函数f定义为求解器f(s,y,C,u,v),则scipy.integrate.odeint解算器起作用f(y,s,C,u,v).为什么会发生这种情况,我该如何解决这个问题呢？

编辑:

Scipy - 版本0.16.0

Answer 1

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

它的发生是因为多年前做出的不幸的API设计决定. odeint并且ode该类需要不同的签名才能解决系统问题.

你可以通过添加一个包装器来修复它,当你使用ode类时,它会改变前两个参数的顺序.例如,你可以改变这个:

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

至

    r = sc.ode(lambda t, x, *args: f(x, t, *args)).set_integrator(solver)

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