python - 是否有任何功能可以在 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 矩阵的求和函数吗？

标签： pythonsympy

Answer 1

您可以通过与相应形状的向量相乘来对向量 A a 求和。

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