从二维 Numpy 数组中提取多个元素的索引，Python

Question

I have one 2D numpy array我有一个 2D numpy 数组

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

I extracted y which is我提取了 y 这是

y = [4 4]

Now I want to extract the Indices of [4 4] in x现在我想在 x 中提取 [4 4] 的索引

I am using the following method我正在使用以下方法

indices = np.argwhere(x == y)

which results [[1 0][2 0]结果[[1 0][2 0]

But in some cases, this code fails.但在某些情况下，此代码会失败。 would you please suggest any other way to perform the same task你能建议任何其他方式来执行相同的任务吗

Answer 1

You can also use nonzero()您也可以使用nonzero()

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