在使用groupby函数对数据进行分组后,我们可以通过遍历每个组来获取组中的记录。groupby函数会返回一个DataFrameGroupBy对象,我们可以通过调用该对象的get_group方法来获取特定组的记录。
示例代码如下:
import pandas as pd
# 创建一个示例数据集
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'Nick'],
'Age': [20, 25, 22, 20, 25],
'Gender': ['Male', 'Male', 'Female', 'Male', 'Male']}
df = pd.DataFrame(data)
# 使用groupby函数进行分组
grouped = df.groupby('Name')
# 遍历每个组并获取记录
for name, group in grouped:
print(f"Group Name: {name}")
print(group)
print()
输出结果如下:
Group Name: John
Name Age Gender
2 John 22 Female
Group Name: Nick
Name Age Gender
1 Nick 25 Male
4 Nick 25 Male
Group Name: Tom
Name Age Gender
0 Tom 20 Male
3 Tom 20 Male
在示例代码中,我们首先使用groupby函数将数据按照“Name”列进行分组,然后使用for循环遍历每个组,并通过group变量获取组中的记录。