如何在Sympy中从解析后的LaTeX方程提取并设置可用于求解的符号？

我已经用SymPy解析了一个LaTeX方程，但现在需要把方程里的变量提取成可直接使用的符号对象，这样SymPy才能针对特定变量求解或重排方程。我知道可以用equation.free_symbols查看方程包含的符号，但不知道怎么自动把这些符号设置成能用于符号求解的对象，比如执行solve(equation,C_L)这种操作。

附代码：

from sympy.parsing.latex import parse_latex
from sympy import init_printing
from sympy.solvers import solve
from sympy import symbols

init_printing() 

equation = parse_latex(r"""
S_{\TO} = \frac{1}{\frac{2g*\rho}{2\left( \WoS \right)}(\mu C_{\L}-C_{\D_{0}}-KC_{\L}^2)}\ln(\frac{(\ToW)-\mu + (\frac{\rho}{2(\WoS)}(\mu C_{\L}-C_{\D_{0}}-KC_{\L}^2))V_{\f}^{2}}{(\ToW)-\mu + (\frac{\rho}{2(\WoS)}(\mu C_{\L}-C_{\D_{0}}-KC_{\L}^{2}))V_{\i}^{2}} )
+ \frac{V_{\TR}^{2}}{0.2g}\sin(\sin^{-1}((\ToW)-\frac{1}{(\LoD)}))
+ \frac{h_{\obstacle}-\frac{V_{\TR}^{2}}{0.2g}(1-\cos(\gamma_{\climb}))}{\tan(\gamma_{\climb})}            
            """)

test = (equation.free_symbols)

其实equation.free_symbols返回的本身就是SymPy的符号对象集合，不是字符串，所以不需要重新定义符号，直接从集合里提取就能用。下面两种方法可以满足需求：

方法一：提取单个目标符号 #

如果你只需要某个特定符号（比如C_L），可以通过遍历集合匹配符号名称来获取：

# 从free_symbols中筛选出名称为'C_L'的符号
C_L = next(sym for sym in equation.free_symbols if sym.name == 'C_L')

# 现在直接用这个符号求解
solution = solve(equation, C_L)

方法二：批量绑定所有符号到当前命名空间 #

如果需要直接使用所有符号（不用每次从集合里取），可以把符号对象批量映射到全局变量：

# 遍历free_symbols，把每个符号绑定到和它名称一致的变量
for sym in equation.free_symbols:
    globals()[sym.name] = sym

# 现在可以直接用所有符号，比如求解C_L
solution = solve(equation, C_L)
# 或者使用其他符号，比如rho
print(rho)

注意事项 #

• 符号的名称要和LaTeX里的定义严格对应，比如LaTeX里的C_{\L}对应SymPy符号的name是'C_L'，可以通过sym.name查看具体名称。
• 批量绑定的时候要注意变量名冲突，如果当前命名空间已经有同名变量，会被覆盖。

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