SymPy中增广矩阵行约简问题：行秩不足时结果不符合预期

在线性代数中，对线性方程组 $Ax=b$ 的增广矩阵 $[A|b]$ 执行行约简是常规操作。当 $b$ 包含符号变量时，约简后的增广列能够直接揭示方程组的兼容性条件（即方程组有解的必要约束）。

• 当系数矩阵 $A$ 满行秩时，SymPy的 rref() 函数返回的结果完全符合预期；
• 但当 $A$ 行秩不足时，rref() 会默认将增广列纳入主元列选择范围，导致过度约简，进而丢失我们需要的兼容性条件。

解决方案：指定主元列范围 #

SymPy的 rref() 函数支持通过 pivots 参数手动指定允许作为主元的列索引（从0开始计数）。我们只需将主元列限制为系数矩阵 $A$ 的列，排除增广列即可。

示例代码

from sympy import Matrix, symbols

# 定义符号变量与增广矩阵
a, b = symbols('a b')
aug_matrix = Matrix([
    [1, 2, 3, a],
    [2, 4, 6, b]
])

# 指定主元列仅为前3列（对应系数矩阵A的列）
reduced_matrix, pivot_cols = aug_matrix.rref(pivots=[0, 1, 2])

print(reduced_matrix)

输出结果

Matrix([[1, 2, 3, a], [0, 0, 0, b - 2*a]])

从约简后的矩阵可以直接读出兼容性条件：$b - 2a = 0$，也就是 $b = 2a$，这正是原方程组有解的必要条件。

原理说明 #

通过手动限定主元列范围，rref() 不会将增广列作为主元进行行变换，因此会保留对应欠秩行的增广列元素，这些元素就是方程组有解的约束条件。

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