python - 如何根据其他数据框进行过滤

Question

这是我的数据 df

Id   Food
1    Apple
1    Banana
2    Banana
2    Cherry
3    Banana
4    DragonFruit

另一个数据集df2

Id
2
4

结果会是

Id   Food
2    Banana
2    Cherry
4    DragonFruit

这是我的代码

df2['one'] = 1
df = df.merge(df2, on='Id', how='left')
df = df[df['one'] == 1]
df= df.drop_duplicates()
df = df[['Id' , 'Food']]

它有效，但我想还有更好的方法

Answer 1

我认为需要isin来自 df2 的 Id 列:

df = df[df['Id'].isin(df2['Id'])]
print (df)
   Id         Food
2   2       Banana
3   2       Cherry
5   4  DragonFruit