在 SymPy 的量子力学模块中，Wavefunction.expr 可直接获取单个波函数的数学表达式，但对 ϕ + ϕ 等线性组合对象（返回 Add 或 Mul 类型）直接调用 .expr 会报错；需递归遍历表达式树，将所有 Wavefunction 实例替换为其 .expr 属性，才能安全提取最终符号表达式。
在 sympy 的量子力学模块中，`wavefunction.expr` 可直接获取单个波函数的数学表达式，但对 `ϕ + ϕ` 等线性组合对象（返回 `add` 或 `mul` 类型）直接调用 `.expr` 会报错；需递归遍历表达式树，将所有 `wavefunction` 实例替换为其 `.expr` 属性，才能安全提取最终符号表达式。
SymPy 的 Wavefunction 类专用于表示量子态的坐标表象（如位置空间波函数），其 .expr 属性设计初衷仅适用于单个 Wavefunction 实例。一旦进行加法（如 ϕ + ϕ）或标量乘法（如 2*ϕ），SymPy 会自动构建标准符号表达式对象（如 Add、Mul），而非保留 Wavefunction 类型——因此组合对象不再具有 .expr 属性，直接访问将触发 AttributeError。
解决该问题的核心思路是：对组合后的符号表达式进行递归遍历，识别并替换所有嵌套的 Wavefunction 子表达式为其 .expr。以下是一个健壮、通用的提取函数：
from sympy import Basic
from sympy.physics.quantum.state import Wavefunction
def extract_wave_expr(expr):
"""
从任意 SymPy 表达式中递归提取所有 Wavefunction 对象的 expr 属性，
返回纯符号表达式（不含 Wavefunction 类型）。
Parameters:
-----------
expr : sympy.Expr
输入表达式，可为 Wavefunction、Add、Mul、Pow 等任意类型
Returns:
--------
sympy.Expr
替换所有 Wavefunction 后的等价符号表达式
"""
if isinstance(expr, Wavefunction):
return expr.expr
elif isinstance(expr, Basic) and hasattr(expr, 'args') and expr.args:
# 递归处理每个子项，并重建当前表达式
new_args = tuple(extract_wave_expr(arg) for arg in expr.args)
return expr.func(*new_args)
else:
return expr
使用示例如下：
import sympy as sp
from sympy.physics.quantum.state import Wavefunction
R = sp.Function('R')
Z = sp.Function('Z')
x, y = sp.symbols('x y', real=True)
ϕ = Wavefunction(R(x)*Z(y), x, y)
combo = ϕ + 2*ϕ # 即 3*R(x)*Z(y)
# ✅ 正确提取
result_expr = extract_wave_expr(combo)
print(result_expr) # 输出: 3*R(x)*Z(y)
print(type(result_expr)) # <class>
# 也可用于更复杂组合
ψ = Wavefunction(sp.sin(x), x)
combined = ϕ + sp.I * ψ
print(extract_wave_expr(combined)) # R(x)*Z(y) + I*sin(x)</class>
⚠️ 注意事项：
- 该函数兼容任意嵌套层级（如 Wavefunction 出现在 sin(ϕ) 或 exp(ϕ + ψ) 中），但需确保 Wavefunction 始终作为表达式树的叶子节点出现；
- 若表达式含非 Wavefunction 的自定义量子态类（如 Ket、Bra），本函数不处理，需按需扩展；
- extract_wave_expr 返回的是纯 sympy.Expr，后续可直接参与符号微分、积分、化简（如 sp.simplify()）等操作；
- 不建议覆盖原始 Wavefunction 类的 __add__ 方法——这会破坏 SymPy 内部表达式结构一致性，且不符合模块设计规范。
综上，extract_wave_expr 提供了一种安全、可复用、符合 SymPy 表达式协议的方式，将量子态波函数组合“降维”为经典符号表达式，是连接量子力学语义与符号计算引擎的关键桥梁。
