python - 查找具有零值的行索引

Question

我有一个 (N,2) 数组代表一些图像坐标。我只想提取两个值为零的行。

例如在这个数组中:

aux = np.array([[0.,-0.0001], [0.0,0.0], [0.0,0.0], [123,0.0]])

我想要一个 numpy 数组作为结果，指示整行都有零:

结果:np.array([1,2])

到目前为止，我正在尝试 where

np.where(aux==0)
(array([0, 1, 1, 2, 2, 3]), array([0, 0, 1, 0, 1, 1]))

但我不明白输出是一个元组。第二个数组是什么？

Answer 1

我想你可以使用 np.all 来做到这一点:

np.all(aux == 0, axis=1)

这将返回一个 bool 数组，其中您的两个值为 0:

array([False,  True,  True, False], dtype=bool)

您可以使用 np.where 提取相应索引的数组(匹配您所需的输出):

np.where(np.all(aux == 0, axis=1))

(array([1, 2]),)