在其元素的总和中替换 sympy 矩阵符号

Question

考虑一个包含矩阵符号元素总和的任意表达式：

>>> from sympy import MatrixSymbol
>>> A = MatrixSymbol('A',3,3)
>>> expr = A[1,2]+A[0,1]+A[2,0]

是否可以简单地用数字矩阵

A

替换矩阵符号

T

？

我假设使用

subs

就足够了，但事实似乎并非如此。

>>> import numpy as np
>>> T = np.arange(9).reshape((3,3))
>>> print(expr.subs(A,T))
A[0, 1] + A[1, 2] + A[2, 0]