在SymPy中对符号矩阵行化简，需使用何种简化函数？

问题描述 #

尝试对包含变量的矩阵执行行化简，代码如下：

from sympy import symbols, Matrix

s, t = symbols('s t')
S = Matrix(2,3, [2,-1, s, 1, -1/2, t])

该矩阵为：

Matrix([[2, -1, s], [1, -0.5, t]])

预期行化简后第二行应为 0 0 t - s/2（即当且仅当s=2t时矩阵奇异），但SymPy返回：

Matrix([[1, -1/2, 0], [0, 0, 1]])

尝试传入simplify=True或simplify=cancel至rref方法均无效，直接传入cancel得到错误结果：

Matrix([[2, 0, s], [1, 0, t]])

解决方案 #

问题根源在于SymPy的rref默认会假设t - s/2 ≠ 0，进而将第二行除以该表达式得到[0,0,1]。要保留t - s/2的原始形式，可通过以下两种方式解决：

方法1：手动执行行变换 #

直接对矩阵执行目标行操作，避免SymPy自动做额外化简：

# 第二行减去第一行的1/2
S_transformed = S.copy()
S_transformed[1, :] = S_transformed[1, :] - S_transformed[0, :] / 2
print(S_transformed)

输出结果：

Matrix([[2, -1, s], [0, 0, t - s/2]])

若需要第一行主元为1，可再将第一行除以2：

S_transformed[0, :] = S_transformed[0, :] / 2
print(S_transformed)

输出：

Matrix([[1, -1/2, s/2], [0, 0, t - s/2]])

方法2：使用rref并指定不做化简 #

调用rref时传入simplify=lambda x: x，阻止SymPy对行变换结果做进一步化简：

rref_result = S.rref(simplify=lambda x: x)
print(rref_result[0])

输出结果：

Matrix([[1, -1/2, s/2], [0, 0, t - s/2]])

说明 #

SymPy的rref默认会对行变换后的表达式进行完全化简，包括将非零行的最后一个元素归一化为1（当该元素为非零表达式时）。通过指定simplify=lambda x: x，可保留行变换后的原始表达式，符合预期的结果形式。

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