数据帧分组到字典 - Python

在Python中，pandas是一种流行的数据分析和处理库。数据帧(DataFrame)是pandas的核心数据结构，代表着表格型数据结构，它可以帮助我们进行数据的管理、整理和处理。在本文中，我们将介绍如何将数据帧中的数据分组到字典中。

分组数据

在进行数据分组之前，我们需要导入pandas库并创建一个数据帧。假设我们有以下数据：

import pandas as pd

# 创建一个关于学生的数据帧
students_data = {'Name': ['John', 'Marry', 'Peter', 'Will', 'Sonia', 'David'], 
                 'Gender': ['Male', 'Female', 'Male', 'Male', 'Female', 'Male'], 
                 'Grade': [85, 78, 92, 90, 88, 95], 
                 'Class': ['A', 'B', 'C', 'A', 'B', 'C']}
df = pd.DataFrame(students_data)

我们现在可以通过调用groupby函数将数据帧中的数据分组。

# 根据班级进行分组
grouped = df.groupby('Class')

这将返回一个DataFrameGroupBy对象，我们可以遍历这个对象以查看每个组的标签和数据。

for name, group in grouped:
    print(name)
    print(group)

输出：

A
   Name Gender  Grade Class
0  John   Male     85     A
3  Will   Male     90     A

B
    Name  Gender  Grade Class
1  Marry  Female     78     B
4  Sonia  Female     88     B

C
    Name Gender  Grade Class
2  Peter   Male     92     C
5  David   Male     95     C

我们可以看到数据被按照班级分成了三组。

将数据分组到字典

我们可以使用字典来存储分组后的结果。以下代码将数据分组到一个字典中。

grouped_dict = {k: v for k, v in grouped}

现在我们可以通过打印字典中的值来查看结果。

print(grouped_dict)

输出：

{'A':    Name Gender  Grade Class
0  John   Male     85     A
3  Will   Male     90     A, 'B':     Name  Gender  Grade Class
1  Marry  Female     78     B
4  Sonia  Female     88     B, 'C':     Name Gender  Grade Class
2  Peter   Male     92     C
5  David   Male     95     C}

我们可以看到数据已经被成功分组到一个字典中了。现在我们可以使用这个字典来查找任何我们想要的数据。

结论

在Python中，我们可以使用pandas库的groupby函数将数据帧中的数据分组到字典中。在分组之后，我们可以使用该字典来轻松地查找任何我们想要的数据。