以下是一个将DataFrame按照某一列的值进行拆分的示例代码:
import pandas as pd
# 创建示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David', 'Ella', 'Frank'],
'Group': ['A', 'B', 'A', 'B', 'B', 'A']}
df = pd.DataFrame(data)
# 按Group列的值进行分组
grouped = df.groupby('Group')
# 将分组后的DataFrame放入一个字典中
groups = {}
for group, data in grouped:
groups[group] = data
# 打印每个分组的DataFrame
for group, data in groups.items():
print(f'Group {group}:')
print(data)
print()
输出结果为:
Group A:
Name Group
0 Alice A
2 Charlie A
5 Frank A
Group B:
Name Group
1 Bob B
3 David B
4 Ella B
在上面的示例中,我们首先创建了一个示例DataFrame,其中包含了两列,Name和Group。然后,我们使用groupby()
函数将DataFrame按Group列的值进行分组。接下来,我们使用一个字典来存储每个分组的DataFrame。最后,我们遍历字典并打印每个分组的DataFrame。