数组之间的交集索引

Question

有没有一种快速的方法可以将数组的每个元素与唯一标识符列表中的每个元素进行比较？

使用 for 循环遍历每个唯一值可以工作，但速度太慢而无法使用。我一直在寻找矢量化解决方案，但没有成功。任何帮助将不胜感激！

arrStart = []

startRavel = startInforce['pol_id'].ravel()

for policy in unique_policies:

    arrStart.append(np.argwhere(startRavel == policy))

样本输入：

startRavel = [1,2,2,2,3,3]

unique_policies = [1,2,3]

示例输出：

arrStart = [[0], [1,2,3],[4,5]]

新数组的长度与唯一值数组的长度相同，但每个元素都是大数组中与该唯一值匹配的所有行的列表。