Python子集与另一个数据帧的数据框

Question

我正在从R转换到Python并尝试在另一个数据框中使用一个字段对数据框进行子集化。什么将是该R代码Python的等效：

final_solution <- subset(df1, item %in% df2$item) 

感谢

Answer 1

我们可以尝试

df1[(df1.item).isin(df2.item)] 

使用重复的例子（与pandas）

import pandas as pd 
df1 = pd.DataFrame({'item' : [1, 2, 3, 4], 
          'fruit' : ['mango', 'apple', 'banana', 'mango']}) 

df2 = pd.DataFrame({'item' : [1, 2]}) 

print(df1[(df1.item).isin(df2.item)]) 

给输出

# fruit item 
#0 mango  1 
#1 apple  2 

Answer 2

假设你使用的是熊猫，你可以merge：

pd.merge(df1, df2, how='inner', on=['df1_col', 'df2_col'])