Python-是否有NumPy函数返回数组中某物的第一个索引?

我知道有一种用于Python列表的方法来返回某些内容的第一个索引:

>>> l = [1, 2, 3]

>>> l.index(2)

1

NumPy数组有类似的东西吗?

回答:

是的,在给定NumPy数组array和值的情况下,这是搜索的答案item

itemindex = numpy.where(array==item)

结果是具有所有行索引,然后是所有列索引的元组。

例如,如果一个数组是二维的,并且它在两个位置包含你的商品,则

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

将等于你的项目,因此

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

以上是 Python-是否有NumPy函数返回数组中某物的第一个索引? 的全部内容, 来源链接: utcz.com/qa/417223.html

回到顶部