Print all set operations between two pandas indices

user_4034025 0
Viewed 24

Is there a more idiomatic/economical way of printing the results of all set operations between two overlapping data frame indices?

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]

I have more than 2 frames and getting all combinations this way is too verbose.

indices pandas set
1 Answers

You could try this:

_ = [
    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()
]

Output:

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