将数据保存到pandas中的多个csv文件

import pandas as pd

# 假设有一个包含多个数据的DataFrame
data = {'Name': ['Tom', 'Nick', 'John', 'Sam'],
        'Age': [20, 25, 30, 35],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# 将数据按照条件分割成多个子集
subsets = []
for city in df['City'].unique():
    subset = df[df['City'] == city]
    subsets.append(subset)

# 保存每个子集到不同的csv文件
for i, subset in enumerate(subsets):
    subset.to_csv(f'data_{i}.csv', index=False)