python - 一次获取 NumPy 数组中多个元素的索引(可以重复)

Question

有没有办法一次性获取 NumPy 数组中多个元素的索引？

例如:

import numpy as np
a = np.array([1, 2, 4])
b = np.array([1, 1, 3, 2, 4])

我想找到a中每个元素在b中的索引，即:[0, 1, 3, 4] .

请注意:

1. b 有重复的值，例如1 此处，方法例如 Getting the indices of several elements in a NumPy array at once 中的方法不起作用，因为它只能找到最左边或最右边的索引，而不是所有索引。因此，如果应用最左边的方法，则使用该方法将得到 [0, 3, 4]。
2. 我想遵守 a 中值的顺序，即结果中的第一个数字对应于 a 中的第一个值，第二个数字代表 a 中的第二个值，依此类推，例如[0, 1] 代表 a 中的 1，[3] 代表 2 code> 在 a 中，而 [4] 代表 a 中的 4，所以答案的顺序是 [0,1,3,4]

Answer 1

感谢我的 friend @rongkaizhang，我在这里发布了他的精彩答案。

np.where(b == np.expand_dims(a, axis=1))[1]