python取交集 dataframe

在Python中，如果你想要从两个DataFrame中获取交集，通常你会使用`pandas`库提供的`intersection`函数，它主要用于列名（columns）的交集，而不是整个DataFrame的交集。如果你想基于某个共享的列来找到两份数据集中匹配的行，可以这样做：

import pandas as pd

# 假设df1和df2是两个DataFrame
df1 = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df2 = pd.DataFrame({'A': [2, 3, 4], 'B': [7, 8, 9]})

# 如果你想找列'A'的交集，可以这样操作：
common_cols = df1.columns.intersection(df2.columns)
if common_cols.any():
    intersection_df = df1[common_cols].merge(df2[common_cols], how='inner')
    print(intersection_df)
else:
    print("两 DataFrame 没有共同的列")

# 如果你想基于某一列的数据交集，例如列'B'，则：
intersection_rows = df1[df1['B'].isin(df2['B'])]
print(intersection_rows)

这里我们先找到两DataFrame共有的列，然后选择这两列创建交集部分。如果需要基于特定值的交集，比如某列的数值，可以使用`.isin()`方法筛选出匹配的行。