我已经附上了我的代码,它是两个身体问题的模型。我曾尝试使用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
答案 0 :(得分:0)
您的初始条件是x0
。您只需要将其从强制2D但非真实x0 = np.empty((6,1))
更改为完全1d x0 = np.empty(6)
。