SymPy为何无法直接求解该波函数的归一化？

问题描述 #

在使用SymPy归一化波函数时，直接调用solve得到的结果是[{t: 0, λ: A**2}]，但指定求解符号A时返回空集合[]，期望得到A=√λ的解。

复现代码 #

from sympy import *

x = Symbol('x', real=True)
t = Symbol('t', real=True)
A = Symbol('A', real=True, positive=True)
λ = Symbol('λ', real=True, positive=True)
ω = Symbol('ω', real=True, positive=True)

# 定义波函数
psi_x_t = A * exp(-λ * Abs(x)) * exp(-I*ω*t)

# 尝试归一化
print(solve(integrate(psi_x_t**2, (x, -oo, oo)) - 1))
print(solve(integrate(psi_x_t**2, (x, -oo, oo)) - 1, A))

原因分析 #

1. 复数波函数的平方误用：波函数psi_x_t是复数形式，直接平方psi_x_t**2得到的是A² * exp(-2λ|x|) * exp(-2Iωt)，属于复数表达式。积分后结果仍为复数，而等式积分结果 - 1 = 0要求复数等于实数，必须满足虚部为0，即exp(-2Iωt)的虚部为0，也就是t=0。
2. 求解约束冲突：当指定求解A时，SymPy需要同时满足复数等式的实部、虚部条件，但t未被固定为0，无法找到符合A为正实数的解，因此返回空列表。

解决方法 #

方法一：使用波函数的模平方（正确归一化方式） #

波函数归一化的标准操作是对模的平方积分，而非直接平方。修改代码如下：

# 计算模平方的积分并归一化
norm_integral = integrate(Abs(psi_x_t)**2, (x, -oo, oo))
print(solve(norm_integral - 1, A))

运行结果：[sqrt(λ)]，完全符合预期。

方法二：先固定t=0再求解A #

如果需要保留直接平方的操作，可先将t代入0，再求解A：

# 代入t=0后计算积分
integral_t0 = integrate(psi_x_t.subs(t, 0)**2, (x, -oo, oo))
print(solve(integral_t0 - 1, A))

运行结果：[sqrt(λ)]。

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