groupby之后部分求和
时间: 2023-08-01 12:08:39 浏览: 99
Java8 stream 中利用 groupingBy 进行多字段分组求和案例
5星 · 资源好评率100%
如果您想在使用`groupby`函数进行分组后,对某些列进行求和,可以使用`agg`函数结合字典来指定每列应用的聚合函数。下面是一个示例:
```python
import pandas as pd
# 创建一个包含多个列的DataFrame对象
df = pd.DataFrame({'Category': ['A', 'B', 'A', 'B', 'A'],
'Value1': [1, 2, 3, 4, 5],
'Value2': [6, 7, 8, 9, 10]})
# 使用groupby函数对Category列进行分组,并对Value1列求和,对Value2列取平均值
result = df.groupby('Category').agg({'Value1': 'sum', 'Value2': 'mean'})
# 打印结果
print(result)
```
在上述示例中,我们首先使用`groupby`函数将DataFrame按照"Category"列进行分组。然后,使用`agg`函数指定每列应用的聚合函数,其中"Value1"列应用了求和函数"sum","Value2"列应用了平均值函数"mean"。最后,将结果打印出来。
您可以根据具体需求,在字典中指定不同的列和相应的聚合函数。
希望对您有所帮助!如果您有任何其他问题,请随时提问。
阅读全文