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

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

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

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

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

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

解决方法

bindices_zero = (array == 0)

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

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

要么

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

（编辑：哈尔滨站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!