我们可以使用pandas
库中的groupby
函数将一个数据框按列进行分割。下面是一个示例代码:
import pandas as pd
# 创建一个示例数据框
data = {'Name': ['John', 'Mike', 'Sarah', 'Tom', 'Amy'],
'Age': [25, 30, 28, 32, 27],
'City': ['New York', 'London', 'Paris', 'Tokyo', 'Sydney']}
df = pd.DataFrame(data)
# 按列选择将数据框分割成多个数据框
grouped_df = df.groupby('City')
# 遍历每个分组并打印分割后的数据框
for city, group in grouped_df:
print("City:", city)
print(group)
print()
输出结果:
City: London
Name Age City
1 Mike 30 London
City: New York
Name Age City
0 John 25 New York
City: Paris
Name Age City
2 Sarah 28 Paris
City: Sydney
Name Age City
4 Amy 27 Sydney
City: Tokyo
Name Age City
3 Tom 32 Tokyo
以上代码使用groupby
函数将数据框按照City
列进行分组,然后遍历每个分组并打印出来。每个分组的结果将作为一个单独的数据框。