在使用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變量獲取組中的記錄。