python错误

Question

嗨，我想用欧拉方法画一个有环的火箭轨道。我的代码是（你可以跳过这个，错误就在下面）：

mo = 1000
m = 900
q = 2.8
u = 3700
vo = 0
xo = 0.0001
g = 9.81
t = np.arange(0,int(1001))
x0 = 0.0001
v0 = 0
N = 1000
at = 1/1000

现在速度和高度：

def v(t):
    return  u*np.log(mo/(mo-q*t))-g*t

def az(t):
    return xo + vo*t - 0.5*g*t**2 + u*t*np.log(mo) + (u/q)*((mo - q*t)*np.log(mo 
    - q*t) + q*t - mo*np.log(mo))

z = xo + vo*t - 0.5*g*t**2 + u*t*np.log(mo) + (u/q)*((mo - q*t)*np.log(mo 
- q*t) + q*t - mo*np.log(mo))

因为这个错误，我把它变成了一个列表：

aj = -g + (q/(mo-q*t))*u
ajj = np.array(aj).tolist()

所以这里是一个循环，我试图用欧拉方法来解析它：

t = np.zeros(N + 1)
t[0] = 0
z = np.zeros(N + 1)
z[0] = x0

v = np.zeros(N + 1)

v[0] = 0


for k in range(0,N):
    v[k + 1] = v[k] +ajj*z[k]*at
    z[k + 1] = z[k] + v[k + 1]*at

它给了我一个错误：

ValueError                                Traceback (most recent call last)
<ipython-input-193-d1513ed52aa0> in <module>()
 11 
 12 for k in range(0,N):
 ---> 13     v[k + 1] = v[k] + ajj*z[k]*at
 14     z[k + 1] = z[k] + v[k + 1]*at
 15 

 ValueError: setting an array element with a sequence.

我知道这意味着什么。这个错误是因为列表ajj，但我不知道如何修复它，以便沿时间绘制z。谢谢你的帮助。抱歉问了这么长的问题.我刚开始编程。你知道吗

Answer 1

您的ODE系统是m'(t)=-q, az'(t)=v(t), v'(t)=u*q/m(t)-g，这里插入了m(t)=m0-q*t。它总是有助于通过ODE函数将系统动力学与解算器分离

def ODEfunc(t,y): az, v = y; return np.array([ v, -g+q*u/(m0-q*t) ]);

这样你就可以按照库克的公式迭代欧拉步骤了

t, y = t+h, y + h*ODEfunc(t,y)