我需要找到数组的零和非零元素的指标.

换句话说,我想找到numpy.nonzero()的补充索引.

我知道这样做的方式如下：

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

然而,这意味着搜索数组两次,这对于大型数组来说效率不高.使用numpy有一种有效的方法吗？

解决方法

假设您已经拥有使用numpy.arange(len(array))的范围,只需获取并存储逻辑索引：

bindices_zero = (array == 0)

那么当你真正需要整数指数时,你可以做到

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

要么

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