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

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

我需要找到数组的零和非零元素的指标. 换句话说,我想找到numpy.nonzero()的补充索引. 我知道这样做的方式如下: indices_zero = numpy.nonzero(array == 0)indices_nonzero = numpy.nonzero(array != 0) 然而,这意味着搜索数组两次,这对于大型数组来说效率不