python groupby 多组分类
时间: 2023-12-07 11:40:09 浏览: 187
20-newsgroups-Text-Classification:使用 20 个新闻组数据集,使用 python 实现文本分类算法
在Python中,groupby函数可以用于对数据进行分组和分组运算。如果需要对多个属性进行分组,可以在groupby函数中传入多个属性,以列表的形式传入即可。例如,下面的代码演示了如何对数据集中的多个属性进行分组:
```python
import pandas as pd
# 创建数据集
data = {'地区': ['北京', '上海', '广州', '北京', '上海', '广州'],
'类型': ['电影', '电影', '电影', '电视剧', '电视剧', '电视剧'],
'评分': [8.5, 9.0, 7.5, 7.8, 8.2, 6.5]}
df = pd.DataFrame(data)
# 对地区和类型进行分组,并计算评分的平均值
result = df.groupby(['地区', '类型'])['评分'].mean()
print(result)
```
上述代码中,我们首先创建了一个数据集,包含了地区、类型和评分三个属性。然后,我们使用groupby函数对地区和类型进行分组,并计算评分的平均值。最后,我们将结果打印出来,输出如下:
```
地区 类型
上海 电影 9.0
电视剧 8.2
北京 电影 8.5
电视剧 7.8
广州 电影 7.5
电视剧 6.5
Name: 评分, dtype: float64
```
可以看到,输出结果中包含了地区、类型和评分三个属性,其中地区和类型组成了一个多级索引,评分则是每个组的平均值。
阅读全文