如何修改SymPy Symbol的字符串表示为合法Python变量名？

有两种简便方式实现你需要的效果：

方法一：自定义Symbol子类 #

直接继承SymPy的Symbol类，重写__str__和__repr__方法，添加自定义字符串表示的参数：

import sympy as sp

class CustomSymbol(sp.Symbol):
    def __new__(cls, latex_name, str_repr=None, **kwargs):
        obj = super().__new__(cls, latex_name, **kwargs)
        obj._str_repr = str_repr if str_repr is not None else latex_name
        return obj
    
    def __str__(self):
        return self._str_repr
    
    def __repr__(self):
        return self._str_repr

# 使用示例
rpj_g0x = CustomSymbol('r_{pj/g0\,x}', str_repr='r_pj_g0x')
print(str(rpj_g0x))  # 输出: r_pj_g0x
print(repr(rpj_g0x)) # 输出: r_pj_g0x
# 不影响LaTeX渲染
print(sp.latex(rpj_g0x)) # 输出: r_{pj/g0\,x}

这个方案完全匹配你的需求，创建Symbol时传入str_repr指定合法Python变量名，同时保留原有的LaTeX格式用于公式渲染。

方法二：自定义字符串转换函数 #

如果不想创建子类，可以用字典映射存储Symbol与目标字符串的对应关系，配合辅助函数实现：

import sympy as sp

# 存储Symbol到自定义字符串的映射表
symbol_str_map = {}

def create_symbol(latex_name, str_repr):
    sym = sp.symbols(latex_name)
    symbol_str_map[sym] = str_repr
    return sym

def sym_str(sym):
    return symbol_str_map.get(sym, str(sym))

# 使用示例
rpj_g0x = create_symbol('r_{pj/g0\,x}', 'r_pj_g0x')
print(sym_str(rpj_g0x))  # 输出: r_pj_g0x
print(sp.latex(rpj_g0x)) # 输出: r_{pj/g0\,x}

这种方式更轻量，但需要调用自定义的sym_str函数获取目标字符串，而非直接使用原生str()。

内容的提问来源于stack exchange，提问作者Nick-H