我知道 Python 列表有一种方法可以返回某物的第一个索引:
>>> xs = [1, 2, 3]
>>> xs.index(2)
1
NumPy 数组有类似的东西吗?
最佳答案
是的,给定一个数组 array
和一个值 item
进行搜索,您可以使用 np.where
如:
itemindex = numpy.where(array == item)
结果是一个元组,首先是所有行索引,然后是所有列索引。
例如,如果一个数组是二维的,并且它在两个位置包含您的项目,那么
array[itemindex[0][0]][itemindex[1][0]]
将等于您的项目,因此将是:
array[itemindex[0][1]][itemindex[1][1]]
关于python - 是否有一个 NumPy 函数来返回数组中某物的第一个索引?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/432112/