python - 有什么方法可以替换 sympy 中的内置参数变量名称吗？

Question

import sympy as sp

a = sp.Matrix([[1,1,1],[0,0,0],[0,0,0]])

b = sp.Matrix([35,0,0])

c = a.gauss_jordan_solve(b)

print(c[0][0])
print(c[0][1])
print(c[0][2])

输出:

-tau0 - tau1 + 35

tau0

tau1

我想将 tau0 和 tau1 替换为任意字母，如何实现？

Answer 1

在 isympy session 中，其中 x 和 y 已定义为符号:

In [22]: sol, params = a.gauss_jordan_solve(b)                                                            

In [23]: sol                                                                                              
Out[23]: 
⎡-τ₀ - τ₁ + 35⎤
⎢             ⎥
⎢     τ₀      ⎥
⎢             ⎥
⎣     τ₁      ⎦

In [24]: params                                                                                           
Out[24]: 
⎡τ₀⎤
⎢  ⎥
⎣τ₁⎦

In [25]: sol.replace(params[0], x).replace(params[1], y)                                                  
Out[25]: 
⎡-x - y + 35⎤
⎢           ⎥
⎢     x     ⎥
⎢           ⎥
⎣     y     ⎦

sol 和 params 是 sympy.Matrix 对象。 taus 是 sympy.Symbol。

或从 gauss_jordan_solve 文档中绘制:

In [74]: sol.xreplace({params[0]:x, params[1]:y})                                                         
Out[74]: 
⎡-x - y + 35⎤
⎢           ⎥
⎢     x     ⎥
⎢           ⎥
⎣     y     ⎦

生成这些tau名称的代码:

    # Free parameters
    # what are current unnumbered free symbol names?
    name = _uniquely_named_symbol('tau', aug,
        compare=lambda i: str(i).rstrip('1234567890')).name
    gen = numbered_symbols(name)
    tau = Matrix([next(gen) for k in range((col - rank)*B_cols)]).reshape(
        col - rank, B_cols)