def integrator(init_state, dt, nstp):
"""
integrate two modes system in the full state sapce.
init_state: initial state [x1, y1, x2, y2]
dt: time step
nstp: number of time step
"""
states = odeint(velocity, init_state, np.arange(0, dt*nstp, dt))
return states
评论列表
文章目录