从NumPy数组中删除一个值

Question

我正在尝试所有只包含NumPy数组中的零的行.例如,我想删除[0,0]从

n = np.array([[1,2], [0,0], [5,6]])

并留下:

np.array([[1,2], [5,6]])

Answer 1

要从numpy表中删除第二行:

import numpy
n = numpy.array([[1,2],[0,0],[5,6]])
new_n = numpy.delete(n, 1, axis=0)

要删除仅包含0的行:

import numpy
n = numpy.array([[1,2],[0,0],[5,6]])
idxs = numpy.any(n != 0, axis=1) # index of rows with at least one non zero value
n_non_zero = n[idxs, :] # selection of the wanted rows