[英]List/Array of matrices in sympy
我想在 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 中实现?
可能有更好的方法。 现在,我设法通过将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))
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.