Sympy中MatrixSymbol与固定尺寸Matrix相乘后元素访问报错问题

当你用MatrixSymbol和Matrix相乘后直接用单索引[0]访问元素时，SymPy无法确定结果矩阵的列数，因此触发IndexError。要得到展开后的乘法表达式，有两种简单方法：

方法一：展开乘积后用双索引访问 #

调用.doit()方法将MatMul对象展开为显式矩阵，再通过双索引[行号, 列号]获取元素：

from sympy import MatrixSymbol, Matrix

a = MatrixSymbol('a', 1, 4)
l = Matrix([1,2,3,4])
# 展开乘积
expanded_product = (a * l).doit()
# 访问1x1矩阵的唯一元素
print(expanded_product[0, 0])

运行后会输出你需要的表达式：a[0, 0]*1 + a[0, 1]*2 + a[0, 2]*3 + a[0, 3]*4

方法二：手动构造元素表达式 #

根据矩阵乘法的定义，直接遍历对应位置的元素相乘求和：

from sympy import MatrixSymbol, Matrix

a = MatrixSymbol('a', 1, 4)
l = Matrix([1,2,3,4])
# 计算第0行0列的元素
element = sum(a[0, i] * l[i] for i in range(4))
print(element)

这种方法也能得到相同的展开结果。

内容的提问来源于stack exchange，提问作者sanket1729