根據與另一個 dataframe 的匹配，在 dataframe 中創建新列

Question

我有兩個數據框：

Dataframe 1：

ID  MONTH
1   2010-01
1   2010-03
1   2010-04
2   2010-01
3   2010-01
3   2010-02

Dataframe 2：

ID  MONTH
1   2010-01
3   2010-02

有沒有辦法根據 dataframe2 中 ID 和 Month 列的行匹配在 Dataframe 1 中創建一個新列？

這樣最終結果將如下所示：

  ID    MONTH Match
0  1  2010-01     Y
1  1  2010-03     N
2  1  2010-04     N
3  2  2010-01     N
4  3  2010-01     N
5  3  2010-02     Y

Answer 1

使用merge + indicator檢查，返回both將為Yes ， left_only將為No

s=df1.merge(df2,indicator=True,how='left')
s['Match']=s.pop('_merge').map({'both':'Y','left_only':'N'})
s
Out[18]: 
   ID    MONTH Match
0   1  2010-01     Y
1   1  2010-03     N
2   1  2010-04     N
3   2  2010-01     N
4   3  2010-01     N
5   3  2010-02     Y