删除2D矩阵中的全零行

Question

是否有高效和/或内置函数来删除2d数组的全零行？我正在看numpy文档，但我还没有找到它。

Answer 1

布尔索引将执行此操作：

In [2]:

a
Out[2]:
array([[4, 1, 1, 2, 0, 4],
       [3, 4, 3, 1, 4, 4],
       [1, 4, 3, 1, 0, 0],
       [0, 4, 4, 0, 4, 3],
       [0, 0, 0, 0, 0, 0]])
In [3]:

a[~(a==0).all(1)]
Out[3]:
array([[4, 1, 1, 2, 0, 4],
       [3, 4, 3, 1, 4, 4],
       [1, 4, 3, 1, 0, 0],
       [0, 4, 4, 0, 4, 3]])

Answer 2

您可以使用内置函数numpy.nonzero。

http://docs.scipy.org/doc/numpy/reference/generated/numpy.nonzero.html