Python中的groupby分组
一、groupby函数
groupby函数功能：对DataFrame进行分组（可单类分组，可多类分组）
需求：按”字段”列对数据data进行分组
groupby函数基本格式：data.groupby([‘分组字段’])
- data：要分组的原始数据
- 分组字段：分组参考的数据列名
举例：
原数据data:
- 单类分组举例
根据”班级”进行分组：
import pandas as pd
data = pd.read_excel('/Users/ABC/Documents/工作簿1.xlsx')
for name, group in data.groupby(['班级']):
num_g = group['班级'].count()
print(name)
print(num_g)
print(group)
print('---------------')
“班级”分组结果：
- 多类分组举例
根据”班级”和”科目”分组：
import pandas as pd
data = pd.read_excel('/Users/ABC/Documents/工作簿1.xlsx')
for name, group in data.groupby(['班级','科目']):
num_g = group['学号'].count()
print(name)
print(num_g)
print(group)
print('---------------')
“班级”和”科目”分组结果：
二、groupby结果对象输出与转型
2.1groupby迭代结果输出
groupby函数产生的结果是个迭代器，若打印输出data.groupby([‘分组字段’])的结果可能会运行处以下结果：
; 解决方法：
要输出具体的结果详情可以用for循环读取（参考上文举例中的代码），其中每个分组结果中包含 分组名称（上文举例代码中：name）和分组记录（上文举例代码中：group）
2.2分组记录- group转化为DataFrame类型
当需要对分组记录- group中的记录进行进一步操作时，发现常用的取列等操作报错
解决方法：
需要将分组记录-group转化为DataFrame类型
转化思路：
用group.values得到一个数组,再将数组用array.tolist()方法转化为列表，再用列表作为参数传入pd.DataFrame()方法中
注意：转化生成的DataFrame没有原数据的列名，需要用DataFrame.columns具体指定
举例：
import pandas as pd
data = pd.read_excel('/Users/ABC/Documents/工作簿1.xlsx')
for name, group in data.groupby(['班级']):
list_group = group.values.tolist()
group_df = pd.DataFrame(list_group)
group_df.columns = ['学号', '姓名', '班级', '科目', '成绩']
print(group_df)
print(group_df['学号'])
print('---------------')
运行结果：
Original: https://blog.csdn.net/O_nice/article/details/124334480
Author: O_nice
Title: Python中的groupby分组
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/670343/
转载文章受原作者版权保护。转载请注明原作者出处！