python - Pandas 根据第二个 df 中的值将列添加到 df

Question

我有两个单独的数据框 df1 和 df2，这两个数据框都包含一个 id 列，该列链接它们之间的行。 df2 有一个 group 列，df1 不包含。我想做的是遍历 df1 中的每个 id 并检查它是否在 df2 中然后是否要采取group 列值并将其放入 df1 中的同名新列下。编写循环遍历的函数是否最简单，或者我可以在这里使用 pandas 技巧吗？

Answer 1

df1 = pd.DataFrame([[1, 'a'],
                    [2, 'b'],
                    [3, 'c']], columns=['id', 'attr'])
df2 = pd.DataFrame([[2, 'd'],
                    [3, 'e'],
                    [4, 'f']], columns=['id', 'group'])

df1.merge(df2, how='left')