如何使用sympy处理矩阵中的表达式？
我正在编写一个库，我可以使用库中的对象构造表达式。 例如， x和y是来自我的库的实例，我可以构造如下表达式：
# below is a simplified version of my class
class MySymbol(object):
import random
_random_value = random.randint(1,4)
def __init__(self, value):
self.value = value
def __add__(self, symbol):
return MySymbol(self.value + symbol.value)
def __mul__(self, symbol):
return MySymbol(self.value * symbol.value)
def __repr__(self):
return str(self.value)
def _get_random_value(self):
return self._random_value
x,y = sympy.symbols('x y')
x = MySymbol(9)
y = MySymbol(3)
import sympy
A = sympy.Matrix([[x,y],[x,y]])
B = sympy.Matrix([[x+y,x*y]])
矩阵运算也是如此。 sympy.Matrix类将这些元素转换为sympy.core.numbers.Integer ，当我希望它们保持其类型MySymbol ：
BA=B*A
print type(BA[0,0])
print type(x*x+y*x+x*x*y) # first element of matrix in *symbolic* form
<class 'sympy.core.numbers.Integer'>
<class '__main__.MySymbol'>
现在，因为BA[0,0]不再是MySymbol的类型，所以我无法调用我想要的方法：
BA[0,0]._get_random_value() # DOES NOT WORK
>> AttributeError: 'Integer' object has no attribute '_get_random_value'
expression = x*x+y*x+x*x*y
expression._get_random_value() # THIS DOES WORK
>> 4
我如何利用sympy.Matrix的矩阵乘法，但仍然允许矩阵的元素保留其类类型的MySymbol ？ 并仍然允许他们的所有方法（如_get_random_value() ）可访问？
您需要从SymPy类继承以在SymPy中使用它。 取决于你的班级在做什么，会告诉你什么类的子类，但最典型的超类是Expr 。 看到我在这里回答一个非常类似的问题。