在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中文網其他相關文章！
