您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

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

5b51 2022/1/14 8:22:35 python 字数 949 阅读 598 来源 www.jb51.cc/python

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

概述

换句话说,我想找到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]

总结

以上是编程之家为你收集整理的python – 获取数组中零和非零元素的索引全部内容,希望文章能够帮你解决python – 获取数组中零和非零元素的索引所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶