多项式元素矩阵

Question

我正在使用NumPy对矩阵进行操作，计算matrixA * matrixB，矩阵的轨迹等等......我的矩阵元素是整数。但我想知道的是，是否有可能使用多项式矩阵。因此，例如我可以处理诸如[x,y;a,b]之类的矩阵，而不是[1,1;1,1]，当我计算跟踪时，它为我提供了多项式x + b，而不是2.在NumPy中是否有一些多项式类哪些矩阵可以使用？

Answer 1

一种选择是使用SymPy Matrices module。 SymPy是Python的符号数学库，它与NumPy可以完全互操作，特别是对于像这样的简单矩阵操作任务。

>>> from sympy import symbols, Matrix
>>> from numpy import trace
>>> x, y, a, b = symbols('x y a b')
>>> M = Matrix(([x, y], [a, b]))
>>> M
Matrix([
[x, y],
[a, b]])
>>> trace(M)
b + x
>>> M.dot(M)
[a*y + x**2, a*b + a*x, b*y + x*y, a*y + b**2]