要遍历Pandas groupby对象的子集,可以使用get_group()
方法。下面是一个示例代码:
import pandas as pd
# 创建一个示例DataFrame
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
'Subject': ['Math', 'Math', 'English', 'English', 'Math'],
'Score': [80, 90, 70, 85, 75]}
df = pd.DataFrame(data)
# 按照姓名进行分组
grouped = df.groupby('Name')
# 遍历每个分组的子集
for name, group in grouped:
print(f"Group: {name}")
print(group)
print()
输出结果:
Group: John
Name Subject Score
2 John English 70
4 John Math 75
Group: Nick
Name Subject Score
1 Nick Math 90
Group: Tom
Name Subject Score
0 Tom Math 80
3 Tom English 85
在这个示例中,首先使用groupby()
方法按照姓名进行分组。然后,通过for
循环遍历每个分组的子集,使用get_group()
方法获取每个分组的数据。
下一篇:遍历 pandas 列