我知道 Python 列表有一种方法可以返回某物的第一个索引:
>>> l = [1, 2, 3] >>> l.index(2) 1
NumPy 数组有类似的东西吗?
是的,给定一个数组 ,array和一个item要搜索的值,您可以np.where用作:
array
item
np.where
itemindex = numpy.where(array==item)
结果是一个元组,首先是所有行索引,然后是所有列索引。
例如,如果一个数组是二维的,并且它在两个位置包含您的项目,那么
array[itemindex[0][0]][itemindex[1][0]]
将等于您的项目,因此将是:
array[itemindex[0][1]][itemindex[1][1]]