numpy - 给定一个子（父子集）numpy数组，在父数组上找到它的索引？

Question

我有两个数组

Parent array (A) = [2,1,2,1,4,4,3]

child array (B) = [1,2,3,4] ( it is actually uniq of A) 

我想使用 numpy 在 A 中为 B 的元素找到索引

在这种情况下，它是

[1,3,0,2,6,4,5]

如果我有这样的字典就好了

{1: [1,3], 2: [0,2], 3:[6], 4: [4,5] }

Answer 1

循环在for这里看起来不错：

out_dict = {}
for a in enumerate(A):
    if a in out_dict:
        out_dict.append(i)
    else:
        out_dict[a] = [i]