如何解决Sympy处理偏微分方程变量分离返回None的问题

pde_separate_mul返回None的核心原因是原始PDE存在大量未化简的冗余代数项，Sympy的分离变量函数无法直接识别可分离结构，按以下步骤处理即可：

步骤1：给符号添加合理物理假设 #

径向坐标$\rho$通常为正实数，添加对应假设后可自动化简绝对值、根号类运算，避免冗余表达式干扰识别。

步骤2：预先化简原始PDE #

原始方程中分子分母存在公共的根号因子可直接约去，同时导数前的复杂系数也可以大幅化简，化简后的方程结构清晰，可被Sympy的分离函数识别。

完整可运行代码 #

import sympy as s

# 给符号添加合理物理假设，ρ为正的径向坐标
m, τ, χ = s.symbols('m τ χ', real=True)
ρ = s.symbols('ρ', real=True, positive=True)
f, F, T = map(s.Function, 'fFT')

# 代入原始方程
de = s.Eq(-m**2*f(τ, ρ, χ) + (2*ρ*s.sqrt(-ρ**2*(1 - ρ**2)/(ρ**2 - 1))*s.Derivative(f(τ, ρ, χ), ρ) + s.sqrt(-ρ**2*(1 - ρ**2)/(ρ**2 - 1))*(ρ**2 - 1)*s.Derivative(f(τ, ρ, χ), (ρ, 2)) + s.sqrt(-ρ**2*(1 - ρ**2)/(ρ**2 - 1))*s.Derivative(f(τ, ρ, χ), (τ, 2))/(1 - ρ**2) + s.sqrt(-ρ**2*(1 - ρ**2)/(ρ**2 - 1))*s.Derivative(f(τ, ρ, χ), (χ, 2))/ρ**2 - s.sqrt(-ρ**2*(1 - ρ**2)/(ρ**2 - 1))*(ρ**2 - 1)**2*(ρ**3*(1 - ρ**2)/(ρ**2 - 1)**2 + ρ**3/(ρ**2 -1) - ρ*(1 - ρ**2)/(ρ**2 - 1))*s.Derivative(f(τ, ρ, χ), ρ)/(ρ**2*(1 - ρ**2)))/s.sqrt(-ρ**2*(1 - ρ**2)/(ρ**2 - 1)), 0)

# 关键：化简原始方程，消去冗余因子
de_simplified = s.simplify(de)

# 再调用分离变量函数
sep_terms = s.pde_separate_mul(de_simplified, f(τ, ρ, χ), [F(ρ, χ), T(τ)])
print(sep_terms)

可选手动分离方案 #

如果需要更灵活的分离操作，可以手动代入分离形式$f=F(\rho,\chi)T(\tau)$，两边除以$f$后整理项，将仅含$\tau$的项移到等式一侧，剩余项放另一侧，两侧相等即等于分离常数，代码示例：

# 手动代入分离形式
de_sub = de_simplified.subs(f(τ, ρ, χ), F(ρ, χ)*T(τ)).doit()
# 两边除以F*T（假设不为0）
de_div = s.simplify(de_sub / (F(ρ, χ)*T(τ)))
# 整理分离τ和其他变量
lhs = de_div.lhs.collect(s.Derivative(T(τ), (τ,2))/T(τ))
# 提取仅含τ的项和剩余项即可得到两个常微分方程

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