下面是一个遍历 pandas 数据框并按数据分组的示例代码:
import pandas as pd
# 创建一个示例数据框
data = {'Name': ['John', 'Mike', 'Sarah', 'John', 'Mike'],
'Age': [28, 32, 30, 28, 32],
'City': ['New York', 'Los Angeles', 'Chicago', 'New York', 'Los Angeles']}
df = pd.DataFrame(data)
# 按 'Name' 列进行分组
grouped = df.groupby('Name')
# 遍历分组并输出每个分组的数据
for name, group in grouped:
print(f"Group: {name}")
print(group)
print('\n')
输出结果如下:
Group: John
Name Age City
0 John 28 New York
3 John 28 New York
Group: Mike
Name Age City
1 Mike 32 Los Angeles
4 Mike 32 Los Angeles
Group: Sarah
Name Age City
2 Sarah 30 Chicago
在这个示例中,我们首先创建了一个示例数据框。然后,我们使用 groupby()
方法按 'Name' 列对数据框进行分组。接下来,我们使用一个循环遍历每个分组,并输出每个分组的数据。