您可以用可变常数求解ode吗？

Question

我有一个单项作业，必须为月球下降创建模型。模型从高处开始，然后向下倾斜放置在两个引擎上。 这是我遇到问题的代码的一部分：

F1 = f*np.array([1, 1, 0.5, 0, 0, 1, 1]) #f-thrust in newtons
F2 = f*np.array([1, 1, 0.5, 0, 0, 1, 1]) #F1 and F2 regime of engines for each time step(percentage of thrust)

t = np.arange(0, np.size(F1), dt) #time domain

#a,b - angles of engines according to global coordinate system
#s0x, v0x, s0y, v0y, fi0, omg0 - initial values
#m - mass
def model_engine(u, t):
    x, vx, y, vy, phi, w = u   
    d1 = vx
    d2 = (1/(m))*(F1*np.cos((a)) - F2*np.cos((b)))
    d3 = vy
    d4 = (1/(m))*(F1*np.sin((a)) + F2*np.sin((b))) + g 
    d5 = w
    d6 = (F1*np.cos(a)*(h/2) - F1*np.sin(a)*(w/2) + F2*np.sin(b)*(w/2) - 
             F2*np.cos(b)*(h/2))/((1/12)*(m)*h**2)
    return np.array([d1, d2, d3, d4, d5, d6])

U = odeint(model_engine, [s0x, v0x, s0y, v0y, fi0, omg0], t)

1. 我不知道如何为定义中的每个时间步实现不同的F1？我得到ValueError: setting an array element with a sequence. 可变质量也是如此，这取决于发动机状态。

2. 解给出了航天器的角度（d6）。以这种方式编写的方程式始终将a和b设为45度，这显然是不正确的，因为航天器旋转。如何从上一步中获取解决方案（角度）并将其放入当前步骤？参见图片以进行澄清

Answer 1

这里是第一个问题的方法：如何实现可变系数？推力仅假定为t的阶跃函数。它是通过在相应的时间插值给定的推力命​​令获得的。插值函数作为附加参数传递给model_engine函数。

模型简化为质量恒定的垂直着陆器。

import matplotlib.pylab as plt

import numpy as np
from scipy.interpolate import interp1d
from scipy.integrate import odeint

from scipy.constants import g

def model_engine(u, t, thrust, mass):
    y, vy = u 

    dydt = vy
    dvy_dt = 1/mass * thrust(t) - g 

    return np.array([dydt, dvy_dt])


# Thrust definition:
F_values = g*np.array([1, 1, 0.5, 0, 0, 2, 2, 1.5, 1]) # f-thrust in newtons
timesteps = np.linspace(0, 10, np.size(F_values)) # timesteps used to defined the thrust

thrust = interp1d(timesteps, F_values, kind='previous', fill_value='extrapolate')

# Solve
s0y, v0y = 102, 0
m0 = 1
t_span = np.linspace(0, 1.2*timesteps[-1], 111)
U = odeint(model_engine, [s0y, v0y], t_span, args=(thrust, m0))

# Graphs
plt.figure();
plt.plot(t_span, U[:, 0], label='altitude')
plt.axhline(y=0, color='black', label='moon surface');
plt.xlabel('time'); plt.legend();

plt.figure();
plt.plot(t_span, thrust(t_span), label='thrust');
plt.xlabel('time'); plt.legend();