python – 获取数组中零和非零元素的索引
发布时间：2020-12-16 23:14:23 所属栏目：Python 来源：网络整理
导读：我需要找到数组的零和非零元素的指标. 换句话说,我想找到numpy.nonzero()的补充索引. 我知道这样做的方式如下： indices_zero = numpy.nonzero(array == 0)indices_nonzero = numpy.nonzero(array != 0) 然而,这意味着搜索数组两次,这对于大型数组来说效率不
|
我需要找到数组的零和非零元素的指标.
换句话说,我想找到numpy.nonzero()的补充索引. 我知道这样做的方式如下： indices_zero = numpy.nonzero(array == 0) indices_nonzero = numpy.nonzero(array != 0) 然而,这意味着搜索数组两次,这对于大型数组来说效率不高.使用numpy有一种有效的方法吗？ 解决方法
假设您已经拥有使用numpy.arange(len(array))的范围,只需获取并存储逻辑索引：
bindices_zero = (array == 0) 那么当你真正需要整数指数时,你可以做到 indices_zero = numpy.arange(len(array))[bindices_zero] 要么 indices_nonzero = numpy.arange(len(array))[~bindices_zero] （编辑：李大同） 【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容! |