python - 使用匹配的 ID 将值从一个数据框填充到另一个数据框

Question

我有两个 pandas 数据框，我想得到 DF1 中每个 ID 的 items_bought 的总和。然后向 DF2 添加一列，其中包含从 DF1 计算出的具有匹配 ID 的 items_bought 的总和，否则用 0 填充它。我怎样才能优雅高效地做到这一点？

DF1

ID | items_bought
1        5
3        8
2        2
3        5
4        6
2        2

DF2

ID
1
2
8
3
2

期望的结果:DF2 成为

ID | items_bought
1        5
2        4
8        0
3        13
2        4

Answer 1

df1.groupby('ID').sum().loc[df2.ID].fillna(0).astype(int)
Out[104]: 
    items_bought
ID              
1              5
2              4
8              0
3             13
2              4

1. 使用 df1 计算每个 ID 的总和。
2. 生成的数据帧现在由 ID 索引，因此您可以通过调用 loc 使用 df2 ID 进行选择。
3. 用 fillna 填空。
4. NA 由 float 类型处理。现在它们已被删除，将列转换回整数。