' ValueError:初始条件y0必须是一维的。'我该如何纠正这个错误?

时间:2017-11-13 08:13:32

标签: python numpy scipy odeint

我已经附上了我的代码,它是两个身体问题的模型。我曾尝试使用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

1 个答案:

答案 0 :(得分:0)

您的初始条件是x0。您只需要将其从强制2D但非真实x0 = np.empty((6,1))更改为完全1d x0 = np.empty(6)