将零数组的索引查找到数组中

Question

我有一个 numpy 数组

my_array = np.array([[1,2,3,4],[5,6,7,8],[0,0,0,0],[1,2,3,4],[0,0,0,0],[0,0,0,1]])

当数组仅包含零值时，我想获取所有索引：

index 2 -> [0,0,0,0]
index 4 -> [0,0,0,0]

存在类似问题的讨论：Find indices of elements equal to zero in a NumPy array

但是在这个解决方案中，我们得到的值等于零，而不是我想要的零数组。

感谢您的帮助。

Answer 1

您可以使用 np.argwhere 和 np.all 来获取所有元素 == 0 的行的索引：

In [11] np.argwhere((my_array == 0).all(axis=1))
Out[11]: 
array([[2],
       [4]], dtype=int64)

或np.nonzero 而不是np.argwhere 会产生更好的输出：

In [12] np.nonzero((my_array == 0).all(axis=1))
Out[12]: (array([2, 4], dtype=int64),)