從二維 Numpy 數組中提取多個元素的索引，Python

Question

我有一個 2D numpy 數組

import numpy as np
x = np.array([[7, 7, 7],
              [4, 7, 7],
              [4, 0, 0]])

我提取了 y 這是

y = [4 4]

現在我想在 x 中提取 [4 4] 的索引

我正在使用以下方法

indices = np.argwhere(x == y)

結果[[1 0][2 0]

但在某些情況下，此代碼會失敗。 你能建議任何其他方式來執行相同的任務嗎

Answer 1

您也可以使用nonzero()

indices = np.nonzero(x == 4)
# (array([1, 2]), array([0, 0]))

Numpy 文檔