温馨提示×

groupby分组后怎么获得组中记录

小亿
617
2023-12-05 14:22:23
栏目: 编程语言

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

0