使用python聚合数据

Question

我有一些数据，我想根据某个字段求和和计数。我的数据是这样的

Value  ID    Object
100    ABD   Type1
200    ABD   Type1
400    ABD   Type2
200    BCE   Type1
100    BCE   Type1
800    JHO   Type3
600    TVM   Type4

我正在尝试计算与 ID 相关的 unique 对象的数量，并将与该 ID 相关的总值相加

ValueSum  ID    CountObject
700       ABD   2  
300       BCE   1
800       JHO   1 
600       TVM   1

我一直在研究的是使用.groupby.() 函数以及.count() 和.sum()，但我似乎无法以正确的格式获取内容。

非常感谢任何帮助。

谢谢！

Answer 1

您可以使用groupby 和agg 传递函数的字典以在您的df 上的多个列上执行：

In [289]:
gp = df.groupby('ID', as_index=False).agg({'Value':sum, 'Object':'nunique'})
gp = gp.rename(columns={'Value':'ValueSum', 'Object':'ObjectCount'})
gp

Out[289]:
    ID  ValueSum  ObjectCount
0  ABD       700            2
1  BCE       300            1
2  JHO       800            1
3  TVM       600            1

这里我们传递一个带有相应列名和要执行的函数的 dict，对于计数，我们使用 nunique 返回唯一值的数量