首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >在NumPy数组中查找等于零的元素的索引

在NumPy数组中查找等于零的元素的索引
EN

Stack Overflow用户
提问于 2011-01-04 05:40:09
回答 9查看 294K关注 0票数 171

NumPy具有有效的函数/方法nonzero()来标识ndarray对象中非零元素的索引。获取值为零的元素的索引的最有效方法是什么?

EN

回答 9

Stack Overflow用户

回答已采纳

发布于 2011-01-04 05:44:00

numpy.where()是我的最爱。

>>> x = numpy.array([1,0,2,0,3,0,4,5,6,7,8])
>>> numpy.where(x == 0)[0]
array([1, 3, 5])
票数 261
EN

Stack Overflow用户

发布于 2017-01-19 10:28:20

那就是np.argwhere

import numpy as np
arr = np.array([[1,2,3], [0, 1, 0], [7, 0, 2]])
np.argwhere(arr == 0)

它以行的形式返回所有找到的索引:

array([[1, 0],    # Indices of the first zero
       [1, 2],    # Indices of the second zero
       [2, 1]],   # Indices of the third zero
      dtype=int64)
票数 45
EN

Stack Overflow用户

发布于 2011-01-04 05:59:07

您可以使用以下命令搜索任何标量条件:

>>> a = np.asarray([0,1,2,3,4])
>>> a == 0 # or whatver
array([ True, False, False, False, False], dtype=bool)

它会将数组作为条件的布尔掩码返回。

票数 27
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/4588628

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档