python - Pandas Groupby 和 Sum Only 一列

Question

所以我有一个数据框 df1，如下所示:

       A      B      C
1     foo    12    California
2     foo    22    California
3     bar    8     Rhode Island
4     bar    32    Rhode Island
5     baz    15    Ohio
6     baz    26    Ohio

我想按列 A 分组，然后对列 B 求和，同时保留列 C 中的值。像这样的:

      A       B      C
1    foo     34    California
2    bar     40    Rhode Island
3    baz     41    Ohio

问题是，当我说

df.groupby('A').sum()

列 C 被移除，返回

      B
A
bar  40
baz  41
foo  34

当我分组和求和时，如何解决这个问题并保留列 C？

Answer 1

这样做的唯一方法是将 C 包含在您的 groupby 中(groupby 函数可以接受一个列表)。

试试这个:

df.groupby(['A','C'])['B'].sum()

另外需要注意的是，如果您需要在聚合之后使用 df，您还可以使用 as_index=False 选项返回一个数据框对象。当我第一次使用 Pandas 时，这给了我一些问题。示例:

df.groupby(['A','C'], as_index=False)['B'].sum()