python - 是否可以在 sympy 中删除矩阵的所有零列和行？

Question

我有一个这样的矩阵:

我想删除零行和零列，在 sympy 中有快速的方法吗？谢谢。

Answer 1

方法row_del 和col_del 允许一次删除一行或一列。看起来您可能有几个，因此一次执行此步骤可能会很慢。

我可能会使用高级索引，传递要保留的行号和列号列表。

A = Matrix([[0, 0, 0, 0], [0, 1, 2, 3], [0, a, b, c], [0, 7, 8, 9]])
m, n = A.shape
rows = [i for i in range(m) if any(A[i, j] != 0 for j in range(n))]
cols = [j for j in range(n) if any(A[i, j] != 0 for i in range(m))]
A = A[rows, cols]

或者，如果您不介意使用 NumPy，这可以更短一些。

import numpy as np
A = Matrix([[0, 0, 0, 0], [0, 1, 2, 3], [0, a, b, c], [0, 7, 8, 9]])
B = np.array(A)
A = Matrix(B[np.ix_(np.any(B != 0, axis=0), np.any(B != 0, axis=1))])