groupby分組后怎么獲得組中記錄

小億
516
2023-12-05 14:22:23

在使用groupby函數(shù)對(duì)數(shù)據(jù)進(jìn)行分組后,我們可以通過(guò)遍歷每個(gè)組來(lái)獲取組中的記錄。groupby函數(shù)會(huì)返回一個(gè)DataFrameGroupBy對(duì)象,我們可以通過(guò)調(diào)用該對(duì)象的get_group方法來(lái)獲取特定組的記錄。

示例代碼如下:

import pandas as pd

# 創(chuàng)建一個(gè)示例數(shù)據(jù)集
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'Nick'],
        'Age': [20, 25, 22, 20, 25],
        'Gender': ['Male', 'Male', 'Female', 'Male', 'Male']}
df = pd.DataFrame(data)

# 使用groupby函數(shù)進(jìn)行分組
grouped = df.groupby('Name')

# 遍歷每個(gè)組并獲取記錄
for name, group in grouped:
    print(f"Group Name: {name}")
    print(group)
    print()

輸出結(jié)果如下:

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函數(shù)將數(shù)據(jù)按照“Name”列進(jìn)行分組,然后使用for循環(huán)遍歷每個(gè)組,并通過(guò)group變量獲取組中的記錄。

0