如何在SymPy中快速删除矩阵的全零行与全零列？

当然可以！在SymPy里处理矩阵的全零行和全零列其实挺简单的，我给你分享两种实用的快速方法：

方法一：手动筛选行和列 #

这种方法直观易懂，适合需要灵活控制的场景，步骤很清晰：

• 先筛选掉全零行：遍历矩阵的每一行，判断是否所有元素都是0，只保留非零行
• 再对处理后的矩阵筛选掉全零列：同理遍历列，保留非零列

代码示例：

from sympy import Matrix

# 创建示例矩阵
mat = Matrix([
    [0, 0, 0],
    [1, 0, 3],
    [0, 0, 0],
    [0, 2, 0]
])

# 第一步：删除全零行
non_zero_rows = mat[[i for i, row in enumerate(mat.rows) if not all(elem == 0 for elem in row)]]
# 第二步：删除全零列
result = non_zero_rows[:, [j for j, col in enumerate(non_zero_rows.cols) if not all(elem == 0 for elem in col)]]

print(result)
# 输出结果：
# Matrix([
# [1, 3],
# [0, 0]])

方法二：使用SymPy内置删除方法（更简洁） #

SymPy的矩阵对象自带row_del()和col_del()方法，我们可以先找出所有全零行/列的索引，然后倒序删除（避免删除后索引变化导致的错误）：

from sympy import Matrix

mat = Matrix([
    [0, 0, 0],
    [1, 0, 3],
    [0, 0, 0],
    [0, 2, 0]
])

# 找出全零行的索引并倒序排列
zero_rows = [i for i, row in enumerate(mat.rows) if all(elem == 0 for elem in row)][::-1]
for i in zero_rows:
    mat.row_del(i)

# 找出全零列的索引并倒序排列
zero_cols = [j for j, col in enumerate(mat.cols) if all(elem == 0 for elem in col)][::-1]
for j in zero_cols:
    mat.col_del(j)

print(mat)
# 输出和方法一完全一致

重要注意事项 #

如果你的矩阵包含符号元素（比如用sympy.symbols定义的变量），直接用elem == 0判断可能会有误差，建议改用elem.is_zero来准确判断元素是否为零：

from sympy import symbols, Matrix
x, y = symbols('x y')
mat = Matrix([[0, x], [0, 0]])
# 用is_zero判断符号元素是否为零
zero_rows = [i for i, row in enumerate(mat.rows) if all(elem.is_zero for elem in row)]

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