1
我試圖解決一個簡單的公式:dM/dr = r*p(r)
在python中。Python odeint,參數
我在r
某些價值p
值:
p(0)=1
,p(1)=3
,p(2)=5
,p(3)=7
,p(4)=9
,p(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