python两个dataframe取交集

可以使用pandas库中的intersection()函数来取两个dataframe的交集。

import pandas as pd

# 创建两个dataframe
df1 = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df2 = pd.DataFrame({'A': [2, 3, 4], 'B': [5, 6, 7]})

# 取两个dataframe的交集
df_intersection = df1.intersection(df2)

print(df_intersection)

输出结果：

   A  B
1  2  5
2  3  6

注意：intersection()函数只能取列名都相同的两个dataframe的交集。如果列名不同，可以先使用rename()函数进行重命名，或者使用merge()函数进行合并后再取交集。