2016-01-31 113 views
1

我試圖解決一個簡單的公式:dM/dr = r*p(r)在python中。Python odeint,參數

我在r某些價值p值:

p(0)=1p(1)=3p(2)=5p(3)=7p(4)=9p(5)=11

我嘗試使用以下的代碼,但我得到的錯誤

由FUNC返回的數組的大小(6)的 Y0大小不匹配(1)。

我認爲問題是我沒有正確匹配p值與r值。應該只有一個初始條件,因爲我只是試圖解決一個方程。任何幫助將不勝感激。

這是我的代碼:

from scipy import integrate 
import numpy as np 

r = np.array([0, 1, 2, 3, 4, 5]) 
p = np.array([1, 3, 5, 7, 9, 11]) 

def deriv (z, r, data): 
    M = r*p 
    return M 

init = np.array([0]) 
soln = integrate.odeint(deriv, init, p, (r,), full_output=True) 
print soln 

回答