python - 按给定条件提取 numpy 行

Question

我的 numpy 数组如下。

import numpy as np


data = np.array([[0,0,0,4],
                [3,0,5,0],
                 [8,9,5,3]])

print (data)

我必须只提取前三个元素不全零的那些行预期结果如下:

result = np.array([[3,0,5,0],
                 [8,9,5,3]])

我尝试过:

res = [l for l in data if l[:3].sum() !=0]
print (res)

它给出了结果。但是，正在寻找更好、更简单的方法。

Answer 1

如果您的数组可以包含负数，

sum 有点不可靠，但 any 始终有效:

result = data[data[:, :3].any(1)]