大熊貓通過比較兩個數據框創建一個新列

Question

我有兩個數據框df1和df2。

df1：

ID     Label
1      a
2      b
5      c

df2：

ID
1
2
3

我想通過比較兩個數據幀在df2中創建一個新列“ label”。 如果ID匹配，則df2中的標簽應等於df1中的標簽。 如果id不在df1中出現，我希望為0。因此最終的df2如下所示：

ID     Label
1      a
2      b
3      0

任何幫助將不勝感激。 謝謝！

Answer 1

您可以使用map ，然后用fillna替換NaN ：

df2['Label'] = df2['ID'].map(df1.set_index('ID')['Label']).fillna(0)
print (df2)
   ID Label
0   1     a
1   2     b
2   3     0