是否有一个Numpy函数返回数组中的第一个索引?

问题:

我知道有一个Python代码返回第一个索引的方法

>>> l = list(1, 2, 3)
>>> l.index(2)
1

有没有像numpy数组那样的东西?

回答:

是的,这里是给出一个Numpy数组,数组和一个值,项目来进行搜索的答案。

itemindex = numpy.where(array==item)

结果是具有第一个所有行索引,然后是所有列索引的元组。
例如,如果数组是两个维,并且它包含您的项目在两个位置,那么

array[itemindex[0][0]][itemindex[1][0]]

将等于你的项目,所以会这样

array[itemindex[0][1]][itemindex[1][1]]

 numpy.where

 
 翻译整理: codewenda.com
英文原文:Is there a Numpy function to return the first index of something in an array?

发表评论

电子邮件地址不会被公开。 必填项已用*标注

44 + = 46