在使用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 Gender2 John 22 FemaleGroup Name: Nick Name Age Gender1 Nick 25 Male4 Nick 25 MaleGroup Name: Tom Name Age Gender0 Tom 20 Male3 Tom 20 Male在示例代码中,我们首先使用groupby函数将数据按照“Name”列进行分组,然后使用for循环遍历每个组,并通过group变量获取组中的记录。

