python - 如何在 sympy 中默认设置 **设置

Question

我有一些方程，我正在使用 sympy 库来打印这些方程，更具体地说是下一行:

print(sp.latex(equation))

所以我想将方程中的“x”替换为“x'_a”，为了实现这一点，我做了这个:

print(sp.latex(equation, symbol_names={x:"x'_a"}))

我的问题是:有没有办法默认设置 symbol_names={x:"x'_a"} ，这样我就不必每次想打印方程时都复制该行。

Answer 1

默认设置可以存储在从 LatexPrinter 创建的自定义打印机中。例如:

import sympy as sp
from sympy.printing.latex import LatexPrinter
x = sp.symbols('x')
myprinter = LatexPrinter({"symbol_names": {x: "x'_a"}})

现在您可以执行 myprinter.doprint(x**2) 并获得 "x'_a^{2}"...顺便说一句，这不是' t 有效的 LaTeX，因此您最好使用名称 {x: "{x'_a}"}。

但实际上，定义打印功能可能更容易。

def mylatex(expr):
    return sp.latex(expr, symbol_names={x: "{x'_a}"})

然后使用mylatex(x**2)。