使用df.groupby写一个实例
时间: 2023-10-24 22:13:45 浏览: 33
好的,下面是一个使用`df.groupby`函数的示例:
假设我们有一个名为`sales`的数据框,其中包含销售数据,包括日期、销售员和销售额。我们想要按日期和销售员对销售额进行分组,并计算每个日期和销售员组的销售总额。我们可以这样做:
``` python
import pandas as pd
# 创建一个示例数据框
sales = pd.DataFrame({
'date': ['2021-01-01', '2021-01-01', '2021-01-02', '2021-01-02'],
'salesperson': ['Alice', 'Bob', 'Alice', 'Bob'],
'sales': [100, 200, 150, 250]
})
# 按日期和销售员分组,并计算销售总额
sales_by_date_sp = sales.groupby(['date', 'salesperson'])['sales'].sum()
# 输出结果
print(sales_by_date_sp)
```
运行上述代码,输出结果如下:
```
date salesperson
2021-01-01 Alice 100
Bob 200
2021-01-02 Alice 150
Bob 250
Name: sales, dtype: int64
```
可以看到,`sales_by_date_sp`是一个序列,其中包含按日期和销售员分组后计算的销售总额。