python - 在 Sympy 索引对象中存储原始索引

Question

我希望能够获取一个索引对象，用值代替它的索引，但仍然记得它的原始索引。

from sympy.tensor import IndexedBase, Idx
C = IndexedBase("C")
i,j = Idx("i"), Idx("j") 
expr = C[i,j]
expr = expr.subs({i:1, j:2})

是否可以从 expr 中获取原始索引 (i,j)？由于表达式是不可变的，我对原始表达式所做的任何修改都不会出现在新表达式中。

我想这里的一般问题是如何将附加信息附加到 Sympy 符号。

Answer 1

您可以将一般信息附加到符号，symbols('x', integer=True)但不能分配值。但是，您可以使用替换符号创建新变量。这使您的原始表达式保持不变：

newexpr = expr.subs({i:1, j:2})

Answer 2

SymPy 在执行subs. 您应该构造您的代码，以便保留原始表达式。