pandas 拆分具有相同结构的 csv

在教程 pandas 合并具有相同结构的 csv 中，我们分享了如何将一个文件夹下很多具有相同结构的 csv 文件合并为一个总的 csv 文件的代码，今天分享的是如何从这个总的 csv 文件还原成原来的一堆子文件，代码如下：

# -*- coding: utf-8 -*-
# author:           inspurer(月小水长)
# create_time:      2022/8/16 14:07
# 运行环境           Python3.6+
# github            https://github.com/inspurer
# website           https://buyixiao.github.io/
# 微信公众号         月小水长

import pandas as pd

import os

input_file = 'all.csv'

output_folder = 'result'
if not os.path.exists(output_folder):
    os.mkdir(output_folder)

df = pd.read_csv(input_file, float_precision='round-trip')
groups = df.groupby(df['origin_file_name'])
for group in groups:
    group[1].drop('origin_file_name', axis=1, inplace=True)
    group[1].to_csv(os.path.join(output_folder, '{}.csv'.format(str(group[0]))), index=False,
                    encoding='utf-8-sig')