numpy：与非零相反（获取零元素的索引）

Question

numpy.nonzero 对于获取非零元素的索引很有用。它返回一个数组元组，每个数组都包含特定轴的非零元素的索引。

除了 zero 元素之外，是否有等价的元素？

对于给定的numpy数组的意思

x = [[1, 2, 0],
     [8, 0, 2],
     [0, 0, 1]]

我可以得到如下索引

ax0_zero_idxs, ax1_zero_idxs = numpy.zero(x)

在哪里

ax0_zero_idxs = [0, 1, 2, 2]
ax1_zero_idxs = [2, 1, 0, 1]

Answer 1

请注意，x == 0 会生成一个 bool 数组，其中包含每个元素 回答问题：x的对应元素是否等于0？

所以你可以将 np.nonzero 应用到这个数组：

result np.nonzero(x == 0)

自己试试吧。

您可以获得满足任何其他条件的元素索引的相同方式。