Pandas分组和求和只针对一列

12 浏览
0 Comments

Pandas分组和求和只针对一列

我有一个名为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？

0