python - 如何使用唯一ID过滤pandas数据框并在其他列上执行计算？

Question

这个问题已经有答案了:

pandas dataframe sum with groupby (1 个回答)

已关闭 4 年前。

我有一个如下所示的数据框。我想过滤我的数据框，以便数据框显示唯一的类型:A、B、C、D。但是，我也想对这些值进行求和。对于类型 A，将为 8+15=13

当前数据框:

ID  type value
1   A    8
2   A    5
3   B    11
4   C    12
5   D    1
6   D    22
7   D    13

预期输出

ID  type value
1   A    13
3   B    11
4   C    12
5   D    36

我是Python新手。带有解释的代码非常感谢!

Answer 1

使用groupby:

print(df.groupby('type',as_index=False)['value'].sum())

如果需要'ID'列:

print(df.groupby('type',as_index=False).agg({'ID':'first','value':'sum'}))