“ValueError:初始条件y0必须是一维的。”如何更正此错误?

2024-05-31 23:58:28 发布

您现在位置:Python中文网/ 问答频道 /正文

我在下面附上了我的代码,这是一个两体问题的模型,我试图用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

Tags: 函数代码模型np向量empty两体mu