python - 获取数组中零和非零元素的索引

Question

我需要找到数组的零元素和非零元素的索引。

换句话说，我想从 numpy.nonzero() 中找到互补索引。

我知道的方法如下:

indices_zero = numpy.nonzero(array == 0)
indices_nonzero = numpy.nonzero(array != 0)

然而，这意味着搜索数组两次，这对于大型数组来说效率不高。是否有使用 numpy 执行此操作的有效方法？

Answer 1

假设你已经有了 numpy.arange(len(array)) 的使用范围，只需获取并存储逻辑索引:

bindices_zero = (array == 0)

然后当你真正需要整数索引时你可以做

indices_zero = numpy.arange(len(array))[bindices_zero]

或

indices_nonzero = numpy.arange(len(array))[~bindices_zero]