不同的scipy ode求解器之間的交換

我有一個求解器，它可以在scipy.integrate.ode和scipy.integrate.odeint之間互換。這是代碼。不同的scipy ode求解器之間的交換

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進行操作？

編輯：

總結的問題了：如果函數f定義爲f(s,y,C,u,v)

scipy.integrate.ode求解器工作，如果函數f定義爲f(y,s,C,u,v)scipy.integrate.odeint解算器的工作原理。爲什麼會發生這種情況，我該如何解決這個問題？

編輯：

SciPy的 - 版本0.16.0

來源

2015-11-04 imranal

如果我互換S和y參數，求解試圖訪問一個浮動，就像它是一個列表。這導致上面的錯誤消息。 – imranal

但是爲什麼你一般需要'''odeint''？ ''odeint'''使用''''ode'''中的''''lsoda'''求解器。 –

對不起，延遲迴復。最初我使用odeint，但是我想增加使用其他求解器的可能性（除了Lsoda）。本例中顯示的代碼只是整個代碼的一小部分。我想我是懶惰的，不想改變代碼。但我也認爲這個問題似乎很重要，因爲這兩個解算器具有不同的變量映射是很奇怪的。 – imranal

爲什麼會發生這種情況，以及如何我，我可以解決這個問題？

這是因爲幾年前做出的一個不幸的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

來源

2015-11-04 14:07:37

謝謝！這就像一個魅力。這樣，我仍然可以使用相同的函數f，而不必在物理上改變參數的順序或創建新函數。 – imranal

不同的scipy ode求解器之間的交換

回答

相關問題