如何找到 NumPy 陣列中元素的第一個索引？

在NumPy 數組中查找項目的第一個索引

在Python 列表中，index() 方法允許我們檢索第一個項目一個元素的出現。 NumPy 能為陣列提供類似的功能嗎？

答案

是的，NumPy 提供了一種方便的方法來定位數組中項目的第一個索引。您可以如下使用 np.where 函數：

import numpy as np

array = np.array([1, 2, 3])
item = 2

itemindex = np.where(array == item)

np.where 函數傳回一個元組，其中包含找到該項目的行索引和列索引。

範例

如果您的陣列是二維的並且包含位於兩個位置的項目，然後：

array[itemindex[0][0]][itemindex[1][0]]  # equals item
array[itemindex[0][1]][itemindex[1][1]]  # also equals item

這使您可以根據第一次出現輕鬆存取所需的元素。

以上是如何找到 NumPy 陣列中元素的第一個索引？的詳細內容。更多資訊請關注PHP中文網其他相關文章！