python - Pandas 获取两个索引之间的交集位置

Question

如果我有 2 个 Pandas 数据帧索引 idx1 和 idx2，如何获取交集的索引位置(从原始索引):

idx1 = pd.Index([1, 2, 3, 4])
idx2 = pd.Index([3, 4, 5, 6])
intersect = idx1.intersection(idx2)

查找 idx1 的 [2, 3] 和 idx 2 的 [0, 1]。

这不起作用:

idx1.get_loc(intersect)

我可以做一个循环来获取这些值，但是有更好的方法吗？

for x in intersect:
    print(idx1.get_loc(x))

Answer 1

如果索引是唯一的，可以使用get_indexer:

idx1.get_indexer(intersect)
# array([2, 3])

idx2.get_indexer(intersect)
# array([0, 1])

如果索引不唯一，则需要使用 isin 方法和 numpy.where:

pd.np.where(idx1.isin(intersect))[0]
# array([2, 3])

pd.np.where(idx2.isin(intersect))[0]
# array([0, 1])