python - 删除零行二维 numpy 数组

Question

我在 numpy 中运行 qr factorization，它返回 ndarrays 列表，即 Q 和 R:

>>> [q,r] = np.linalg.qr(np.array([1,0,0,0,1,1,1,1,1]).reshape(3,3))

R 是一个二维数组，底部有零线(甚至在我的测试集中证明了所有示例):

>>> print r
[[ 1.41421356  0.70710678  0.70710678]
 [ 0.          1.22474487  1.22474487]
 [ 0.          0.          0.        ]]

。现在，我想将 R 分成两个矩阵 R_~:

[[ 1.41421356  0.70710678  0.70710678]
 [ 0.          1.22474487  1.22474487]]

和R_0:

[[ 0.          0.          0.        ]]

(提取所有零线)。它似乎接近这个解决方案:deleting rows in numpy array .

编辑:
更有趣的是:np.linalg.qr() 返回一个 n x n-矩阵。不是，如我所料:

A := n x m
Q := n x m
R := n x m

Answer 1

将 np.all 与 axis 参数一起使用:

>>> r[np.all(r == 0, axis=1)]
array([[ 0.,  0.,  0.]])
>>> r[~np.all(r == 0, axis=1)]
array([[-1.41421356, -0.70710678, -0.70710678],
       [ 0.        , -1.22474487, -1.22474487]])