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]

Answer 2

您可以使用布尔索引：

In [82]: a = np.random.randint(-5, 5, 100)

In [83]: a
Out[83]: 
array([-2, -1,  4, -3,  1, -2,  2, -1,  2, -1, -3,  3, -3, -4,  1,  2,  1,
        3,  3,  0,  1, -3, -4,  3, -5, -1,  3,  2,  3,  0, -5,  4,  3, -5,
       -3,  1, -1,  0, -4,  0,  1, -5, -5, -1,  3, -2, -5, -5,  1,  0, -1,
        1,  1, -1, -2, -2,  1,  1, -4, -4,  1, -3, -3, -5,  3,  0, -5, -2,
       -2,  4,  1, -4, -5, -1,  3, -3,  2,  4, -4,  4,  2, -2, -4,  3,  4,
       -2, -4,  2, -4, -1,  0, -3, -1,  2,  3,  1,  1,  2,  1,  4])

In [84]: mask = a != 0

In [85]: a[mask]
Out[85]: 
array([-2, -1,  4, -3,  1, -2,  2, -1,  2, -1, -3,  3, -3, -4,  1,  2,  1,
        3,  3,  1, -3, -4,  3, -5, -1,  3,  2,  3, -5,  4,  3, -5, -3,  1,
       -1, -4,  1, -5, -5, -1,  3, -2, -5, -5,  1, -1,  1,  1, -1, -2, -2,
        1,  1, -4, -4,  1, -3, -3, -5,  3, -5, -2, -2,  4,  1, -4, -5, -1,
        3, -3,  2,  4, -4,  4,  2, -2, -4,  3,  4, -2, -4,  2, -4, -1, -3,
       -1,  2,  3,  1,  1,  2,  1,  4])

In [86]: a[-mask]
Out[86]: array([0, 0, 0, 0, 0, 0, 0])

Answer 3

我不确定用于完成此任务的内置 numpy 方法，但我相信您可以使用老式for循环。就像是：

indices_zero = []
indices_nonzero = []

for index in xrange(len(array)):
    if array[index] == 0:
        indicies_zero.append(index)
    else:
        indicies_nonzero.append(index)

这样的事情应该完成你想要的，只循环一次。