Numpy选择非零行

Question

我想只选择没有任何0元素的行.

data = np.array([[1,2,3,4,5],
                [6,7,0,9,10],
                [11,12,13,14,15],
                [16,17,18,19,0]])

结果将是:

array([[1,2,3,4,5],
       [11,12,13,14,15]])

Answer 1

用途numpy.all:

>>> data[np.all(data, axis=1)]
array([[ 1,  2,  3,  4,  5],
       [11, 12, 13, 14, 15]])