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

找出陣列元素的第一個索引

與Python 清單類似，NumPy 陣列提供了一種定位特定元素第一次出現的方法價值。這個問題深入研究了 list.index() 方法的 NumPy 等效項。

NumPy 的索引檢索方法

檢索 NumPy 數組中第一個元素的索引匹配給定值，使用 np.where函數：

itemindex = np.where(array == item)

這個函數傳回包含兩個陣列的元組：

• 第一個陣列包含符合元素的行索引。
• 第二個陣列包含符合元素的欄位索引。

此元組提供了存取數組中匹配元素所需的座標。

範例用法

考慮一個二維陣列：

array = np.array([[1, 2, 3], [4, 5, 6]])

如果我們想找出值5 的第一次出現，我們可以使用：

itemindex = np.where(array == 5)
print(itemindex[0][0], itemindex[1][0])  # Output: 1 1

這表示值5位於數組內的行索引1和列索引1處，可以確認作者：

print(array[itemindex[0][0]][itemindex[1][0]])  # Output: 5

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