下面是一个示例,展示了如何使用 Pandas 按条件分组数据:
import pandas as pd
# 创建示例数据
data = {'Name': ['John', 'Mike', 'Sarah', 'Mike', 'John'],
'Age': [27, 30, 25, 30, 26],
'City': ['New York', 'Los Angeles', 'Chicago', 'Los Angeles', 'New York']}
df = pd.DataFrame(data)
# 按条件分组数据
grouped_data = df.groupby('City')
# 遍历每个分组
for city, group in grouped_data:
print("City:", city)
print(group)
print('\n')
输出结果:
City: Chicago
Name Age City
2 Sarah 25 Chicago
City: Los Angeles
Name Age City
1 Mike 30 Los Angeles
3 Mike 30 Los Angeles
City: New York
Name Age City
0 John 27 New York
4 John 26 New York
上述代码首先创建了一个包含姓名、年龄和城市的示例数据。然后,使用 groupby
方法将数据按城市进行分组。最后,通过遍历每个分组,可以访问每个分组的数据。
上一篇:按条件分组