我在下面附上了我的代码,这是一个两体问题的模型,我试图用1D向量作为odeint函数的参数,但是我仍然面临错误。非常感谢您的建议。在
tspan = np.linspace(0,p,(p/60.0)+1.0)
x0 = np.empty((6,1))
x0[0]=r[0]
x0[1]=r[1]
x0[2]=r[2]
x0[3]=v[0]
x0[4]=v[1]
x0[5]=v[2]
tspan = np.linspace(0,p,(p/60.0)+1.0)
x = odeint(ot.odetwobody,x0,tspan)
def odetwobody(x,t):
mu = 398600.0
r = np.sqrt(x[1]**2+x[2]**2+x[3]**2)
xddot = np.empty((6,1))
xddot[0] = x[3]
xddot[1]=x[4]
xddot[2]=x[5]
xddot[3]=(-mu*x[0])/r**3
xddot[4]=(-mu*x[1])/r**3
xddot[5]=(-mu*x[2])/r**3
return xddot
您的初始条件是
x0
。您只需将其从强制2D改为完全1dx0 = np.empty((6,1))
,而不是真正的x0 = np.empty(6)
。在相关问题 更多 >
编程相关推荐