阅读量:7
在使用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变量获取组中的记录。