我需要找到数组的零和非零元素的指标.
换句话说,我想找到numpy.nonzero()的补充索引.
我知道这样做的方式如下：
indices_zero = numpy.nonzero(array == 0) indices_nonzero = numpy.nonzero(array != 0)
解决方法
假设您已经拥有使用numpy.arange(len(array))的范围,只需获取并存储逻辑索引：
bindices_zero = (array == 0)
那么当你真正需要整数指数时,你可以做到
indices_zero = numpy.arange(len(array))[bindices_zero]
要么
indices_nonzero = numpy.arange(len(array))[~bindices_zero]