打印两个 pandas 索引之间的所有集合操作

Question

是否有更惯用/更经济的方法来打印两个重叠数据帧索引之间所有集合操作的结果？

import pandas as pd

df1 = pd._testing.makeMixedDataFrame()
df2 = df1.copy()
df2.index = range(3, 8)

intersec = list(df1.index.intersection(df2.index))
diff12 = list(df1.index.difference(df2.index))
diff21 = list(df2.index.difference(df1.index))

print(f"{intersec = }") # intersec = [3, 4]
print(f"{diff12 = }") # diff12 = [0, 1, 2]
print(f"{diff21 = }") # diff21 = [5, 6, 7]

我有超过 2 个帧，用这种方式获取所有组合太冗长了。

Answer 1

你可以试试这个：

_ = [
    print(f"{k} = {v.index.to_list()}")
    for k, v in {
        "intersec": df1[df1.index.isin(df2.index)],
        "diff12": df1[~df1.index.isin(df2.index)],
        "diff21": df2[~df2.index.isin(df1.index)],
    }.items()
]

输出：

intersec = [3, 4]
diff12 = [0, 1, 2]
diff21 = [5, 6, 7]