sympy 中的矩陣列表/數組

Question

我想在 sympy 中創建以下表達式：

M 、 U和A^(k)是大小為m方陣。

到目前為止，我設法為給定的p做到了這一點：

import sympy

m = sympy.symbols('m', integer=True)
p = 3

A = [sympy.MatrixSymbol('A^({})'.format(k), m, m) for k in range(p)]
M = sympy.MatrixSymbol('M', m, m)
U = sympy.MatrixSymbol('U', m, m)

expr = M*A[0]*U
for k in range(1, p):
    expr += M*A[k]*U
expr

但是，我不想將p設置為固定數字。 相反，我希望p成為一個符號，就像m一樣。 這如何在 sympy 中實現？

Answer 1

可能有更好的方法。 現在，我設法通過將A定義為函數而不是列表來實現我想要的：

m, p, k = symbols('m, p, k')

M = sympy.MatrixSymbol('M', m, m)
U = sympy.MatrixSymbol('U', m, m)

class A(sympy.Function):
    @classmethod
    def eval(cls, k):
        return sympy.MatrixSymbol('A^({})'.format(k), m, m)

sympy.Sum(M * A(k) * U, (k, 1, p))