python中groupby的用法
时间: 2023-12-07 21:37:12 浏览: 75
在Python中,groupby函数是一个非常有用的函数,它可以根据指定的键对数据进行分组,并将分组后的数据进行聚合处理。下面是一个简单的例子来演示groupby函数的用法:
假设我们有一个学生列表,其中包含每个学生的姓名、班级和成绩。我们想要根据班级对学生进行分组,并计算每个班级的平均成绩。代码如下:
```python
import pandas as pd
# 创建学生列表
students = {
'name': ['Alice', 'Bob', 'Charlie', 'David', 'Emily', 'Frank', 'Grace', 'Helen'],
'class': ['A', 'B', 'A', 'B', 'A', 'B', 'A', 'B'],
'score': [80, 90, 70, 60, 85, 95, 75, 65]
}
# 将学生列表转换为DataFrame
df = pd.DataFrame(students)
# 根据班级分组,并计算平均成绩
result = df.groupby('class')['score'].mean()
# 输出结果
print(result)
```
输出结果为:
```
class
A 78.333333
B 80.000000
Name: score, dtype: float64
```
这里我们使用了pandas库中的DataFrame数据结构和groupby函数。首先,我们将学生列表转换为DataFrame,然后使用groupby函数根据班级对学生进行分组,并计算每个班级的平均成绩。最后,我们将结果输出到控制台。
阅读全文