python - 获取另一个 numpy 数组中一个 numpy 数组的索引

Question

这个问题已经有答案了:

Finding indices of matches of one array in another array (4 个回答)

已关闭 4 年前。

我正在尝试获取另一个 numpy 数组内的 numpy 数组的索引，例如，我有

 y = array([1, 2, 3, 4, 5, 8])
 x = np.array([3, 1, 8]) 

x 包含在 y 中，所以我想要得到的是索引 idx x 数组在 y 数组上的顺序相同，所以这里是 idx = array([2, 0, 5])这样我们就有 np.array_equal(y[idx] ,x)产量True ,我尝试使用 np.argwhere(np.in1d(y,x)) ，但显然我没有得到相同的顺序，我知道我总是可以使用列表理解 idx = [ list(y).index(el) for el in x] ，但我更喜欢使用 numpy。有什么想法吗？

Answer 1

来自Is there a NumPy function to return the first index of something in an array?

>>> t = array([1, 1, 1, 2, 2, 3, 8, 3, 8, 8])
>>> nonzero(t == 8)
(array([6, 8, 9]),)
>>> nonzero(t == 8)[0][0]
6

因此请根据您的情况进行调整:

lst=[]
for item in x:
    lst.append( list(nonzero(y == item)) )