Python利用df.groupby实现表格的分组

目标：利用Python pandas库的df.groupby将Excel表格按照班级进行分组管理

测试的数据如下：

代码如下：

import pandas as pd

df = pd.read_excel('test.xlsx')

data_group = df.groupby(['组别'])  # 将数据按“组别”，也就是按班级进行分组
Class1 = data_group.get_group('一班')  #获取组别为一班的数据
Class2 = data_group.get_group('二班')
Class3 = data_group.get_group('三班')

Class1_and_Class2 = pd.concat([Class1, Class2], ignore_index=True)# 合并一班和二班的数据

结果测试

打印“一班”的数据

print(Class1)

 

打印“一班”和“二班”合并后的数据

print(Class1_and_Class2)

 

将表格保存为Excel文件（以一班数据为例）

Class1.to_excel('Class1.xlsx')