Numpy:如何快速找到第一个价值指数?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (29)

如何在Numpy数组中找到数字第一次出现的索引?速度对我很重要:

itemindex = numpy.where(array==item)[0][0]
nonzero(array == item)[0][0]

提问于
用户回答回答于

可以:

@jit(nopython=True)
def find_first(item, vec):
    """return the index of the first occurence of item in vec"""
    for i in xrange(len(vec)):
        if item == vec[i]:
            return i
    return -1

然后:

>>> a = array([1,7,8,32])
>>> find_first(8,a)
2

扫码关注云+社区