是否有任何函数可以对Sympy上的Matrix求和？

Question

当我对矩阵求和时：

import sympy
i = sy.symbols('i')
A=Matrix([1,i,3,3*i])
summation(A[i],(i,0,3))

结果：

IndexError: Invalid index a[i]

我知道我可以通过循环解决此问题。

没有任何求和的函数可以支持sympy矩阵吗？

Answer 1

您可以将向量A a与各个形状的向量相乘来求和。

A.T * ones(A.shape[0], 1)