使用scipy odeint求解ODE时遇复数转换失败问题求助

直接报错原因：参数顺序不匹配 #

你的报错核心原因是**lambdify生成的函数调用时参数顺序和定义时完全不一致**，导致符号变量未被正确替换为数值，返回的是SymPy表达式而非浮点数，进而触发"无法转换表达式为浮点数"的错误。

看你的代码：

• dQ1dt_f的参数定义顺序是：(mx, my, mp, j, k1, k2, k3, k4, k5, r, b1, b2, f1, f2, Bp, tau, x, x_d, y, y_d, P1, P1_d, ..., Signum_x, Signum_y, ...)
• 但调用时你把Signum系列参数提前到了x,x_d前面，这种错位会导致原本应该传入数值的位置传入了Signum相关的符号（或反之），函数返回未完全求值的SymPy表达式，而非浮点数，最终引发报错。

复数出现的可能原因 #

如果修正参数顺序后仍然出现复数，可能有两种情况：

1. ODE系统本身的特性：比如你的系统包含复特征值（常见于振荡或不稳定的线性/非线性系统），即使初始条件是实数，求解过程中也可能出现复数值。
2. 符号推导错误：在二阶ODE转一阶、或者求解sols[x_dd]的过程中存在代数错误，导致表达式中引入了复数项（比如平方根下出现负数）。

解决方案步骤 #

1. 修正参数调用顺序 #

严格按照lambdify定义的参数顺序传参，把Signum系列参数放到最后：

# 修正后的dQ1dt_f调用
dQ1dt_f(mx, my, mp, j, k1, k2, k3, k4, k5, r, b1, b2, f1, f2, Bp, tau,
        x, x_d, y, y_d,
        P1, P1_d, P2, P2_d,
        P3, P3_d, P4, P4_d, P5, P5_d, P6, P6_d,
        P7, P7_d, P8, P8_d, P9, P9_d, P10, P10_d,
        Signum_x, Signum_y, Signum_P1, Signum_P2, Signum_P3, Signum_P4,
        Signum_P5, Signum_P6, Signum_P7, Signum_P8, Signum_P9, Signum_P10)

可以单独测试dQ1dt_f的返回值，确认它返回的是浮点数而非SymPy表达式：

# 测试单个函数调用
test_val = dQ1dt_f(1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,
                   0,0,0,0,
                   0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
                   1,1,1,1,1,1,1,1,1,1)
print(type(test_val))  # 应该输出<class 'float'>或<class 'numpy.float64'>

2. 处理复数问题的替代方案 #

如果修正参数后仍然出现复数，可以尝试以下方法：

• 使用支持复数的求解器：改用scipy.integrate.solve_ivp，它原生支持复数值状态变量。只需要将初始条件转为复数类型：

  from scipy.integrate import solve_ivp
  
  S0 = np.array([0,0,...], dtype=np.complex128)  # 转成复数数组
  t_span = [0,40]
  t_eval = np.linspace(0,40,1001)
  sol = solve_ivp(dSdt, t_span, S0, t_eval=t_eval, method='LSODA')
  

• 检查符号推导正确性：重新验证二阶ODE转一阶的过程，以及sols[x_dd]的求解是否正确，确保没有引入不必要的复数项（比如错误的平方根、分母为零的情况）。
• 调整求解器参数：对于odeint，可以尝试调小rtol和atol参数，提高数值精度：

  ans = odeint(dSdt, y0=S0, t=t, rtol=1e-8, atol=1e-10)
  

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