如何使用Pandas按国家分组生成对应CSV文件？

可以用pandas的分组功能轻松实现这个需求，具体操作如下：

• 确保已导入pandas并加载好目标DataFrame（假设你的DataFrame变量名为df）
• 通过groupby('Country')按国家对数据分组
• 遍历每个分组，将每组数据导出为以对应国家命名的CSV文件

完整代码示例 #

import pandas as pd

# 若已加载DataFrame可跳过此初始化部分
data = {
    'Country': ['France', 'France', 'Switzerland', 'United States'],
    'Region': ['Pas de Calais', 'Provence', 'Geneva', 'Florida'],
    'Population': [500000, 200000, 400000, 1200000]
}
df = pd.DataFrame(data)

# 按国家分组并导出CSV
for country, group in df.groupby('Country'):
    # 以国家名作为CSV文件名，比如France.csv
    group.to_csv(f'{country}.csv', index=False)

代码说明 #

• groupby('Country')会将原DataFrame拆分为多个子DataFrame，每个子DataFrame对应一个国家的全部数据
• 遍历过程中，country是当前分组的国家名称，group是该国家对应的数据集
• to_csv(f'{country}.csv', index=False)用于保存分组数据，index=False表示不导出索引列，让输出的CSV结构更简洁

执行代码后，当前目录会生成France.csv、Switzerland.csv和United States.csv三个文件，每个文件仅包含对应国家的所有数据。

内容的提问来源于stack exchange，提问作者mollieda