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