Pandas 中两个数据框的交集 – Python

使用预定义函数merge()可以轻松计算 Pandas 中两个数据帧的交集。此函数将两个数据帧作为参数并返回它们之间的交集。

句法：

pd.merge(df1, df2, how)

示例 1：

import pandas as pd
  
# Creating Data frames
df1 = {'A': [1, 2, 3, 4],
         'B': ['abc', 'def', 'efg', 'ghi']} 
df2 = {'A': [1, 2, 3, 4 ],
         'B': ['Geeks', 'For', 'efg', 'ghi'],
         'C':['Nikhil', 'Rishabh', 'Rahul', 'Shubham']} 
           
d1 = pd.DataFrame(df1)
d2 = pd.DataFrame(df2) 
  
# Calling merge() function
int_df = pd.merge(d1, d2, how ='inner', on =['A', 'B'])
print(int_df)

输出：

A    B        C
0  3  efg    Rahul
1  4  ghi  Shubham

示例 2：

import pandas as pd
  
# Creating Data frames
df1 = {'A': [1, 2, 3, 4],
         'B': ['Geeks', 'For', 'efg', 'ghi']} 
df2 = {'A': [1, 2, 3, 4 ],
         'B': ['Geeks', 'For', 'abc', 'cde'],
         'C':['Nikhil', 'Rishabh', 'Rahul', 'Shubham']} 
           
d1 = pd.DataFrame(df1)
d2 = pd.DataFrame(df2) 
  
# Calling merge() function
int_df = pd.merge(d1, d2, how='inner', on=['A', 'B'])
print(int_df)

输出：

A      B        C
0  1  Geeks   Nikhil
1  2    For  Rishabh