如何快速判断scipy.solve_ivp()中触发的事件类型？

下面是几种直接可行的方法，帮你区分是第一根还是第二根摆杆的翻转事件被触发：

• 给事件函数添加唯一标识返回值
  不要让事件函数只返回触发条件的布尔值，改为返回包含触发状态和事件标识的结构。以Python场景为例：

  import numpy as np
  
  def event_pendulum1(t, y):
      # 第一根摆杆翻转的判断逻辑（示例：角度超过±π）
      flip_condition = (y[0] >= np.pi) or (y[0] <= -np.pi)
      # 返回(触发状态, 事件标识)
      return (flip_condition, 1) if flip_condition else (False, 0)
  
  def event_pendulum2(t, y):
      flip_condition = (y[2] >= np.pi) or (y[2] <= -np.pi)
      return (flip_condition, 2) if flip_condition else (False, 0)
  

  求解器返回的事件结果中，你可以直接提取标识值，1对应第一根摆杆，2对应第二根。

• 利用求解器的事件传入顺序匹配索引
  如果你用SciPy这类支持多事件的求解库，传入事件函数的顺序会和触发结果的索引一一对应。比如先传第一根的事件函数，再传第二根的：

  from scipy.integrate import solve_ivp
  
  # 假设已定义系统微分方程your_ode_func、时间区间t_span和初始值y0
  sol = solve_ivp(your_ode_func, t_span, y0, events=[event_pendulum1, event_pendulum2])
  
  # 遍历触发的事件，找到非空结果对应的索引
  for idx, t_event in enumerate(sol.t_events):
      if len(t_event) > 0:
          print("第一根摆杆发生翻转" if idx == 0 else "第二根摆杆发生翻转")
  

• 在事件函数内部设置触发标记
  定义一个全局变量或者类属性，当事件触发时直接赋值对应的标识：

  import numpy as np
  
  triggered_event = None
  
  def event_pendulum1(t, y):
      global triggered_event
      flip_condition = (y[0] >= np.pi) or (y[0] <= -np.pi)
      if flip_condition:
          triggered_event = "pendulum1_flip"
      return flip_condition
  
  def event_pendulum2(t, y):
      global triggered_event
      flip_condition = (y[2] >= np.pi) or (y[2] <= -np.pi)
      if flip_condition:
          triggered_event = "pendulum2_flip"
      return flip_condition
  

  事件触发后，直接读取triggered_event的值就能明确是哪个事件被触发。

内容的提问来源于stack exchange，提问作者user1254621