Python求解含复数ODE时遇警告且结果错误，求修复方案

核心问题 #

scipy.integrate.odeint默认仅支持实数域的常微分方程求解。你的初始条件y0 = [0,1]是实数类型，但导数函数func返回的是复数结果，odeint会强制将复数转换为实数并丢弃虚部——这就是你看到ComplexWarning的原因，也是计算结果不符合预期的根源。

修复步骤 #

1. 将初始条件改为复数类型：把y0从实数列表改为复数数组，明确告诉odeint要处理复数ODE。
2. （可选）确认导数函数的计算逻辑匹配你的物理模型（当前函数形式是典型的拉比振荡模型，逻辑无明显问题）。

修改后的完整代码 #

import numpy as np
from matplotlib import pyplot as plt
from scipy.integrate import odeint

exp = np.exp
pi = np.pi

def func(y, t, W, omega, delta, omega_Rabi):
    c_up, c_down = y
    dydt = [
        -1j * np.exp(-1j * delta * t) * (1j * W + omega_Rabi) * c_down,
        -1j * np.exp(1j * delta * t) * (-1j * W + omega_Rabi) * c_up
    ]
    return dydt

W = 2*pi*10
omega_Rabi = 2*pi*300
delta = 2*pi*5000
omega = 2*pi*100000
# 将初始条件改为复数类型，明确指定复数 dtype
y0 = np.array([0, 1], dtype=np.complex128)

t = np.linspace(0, 0.01, 100)
sol = odeint(func, y0, t, args=(W, omega, delta, omega_Rabi))

plt.plot(t, abs(sol[:, 0])**2, 'b')
plt.xlabel('Time')
plt.ylabel('Population of upper state')
plt.show()

修复原理 #

当y0是复数类型时，odeint会自动切换到复数模式进行积分，全程保留虚部信息，不会再触发类型转换警告，计算结果也会符合拉比振荡的预期（比如能看到上态布居随时间周期性振荡）。

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