按行数拆分CSV文件并保留头部
创始人
2024-08-22 06:00:55
0

下面是一个示例代码,可以按行数拆分CSV文件并保留头部:

import csv

def split_csv(input_file, output_path, rows_per_file):
    with open(input_file, 'r') as file:
        reader = csv.reader(file)
        header = next(reader)  # 读取头部

        file_index = 1
        row_count = 0
        output_file = None

        for row in reader:
            if output_file is None or row_count >= rows_per_file:
                if output_file is not None:
                    output_file.close()
                output_file = open(f'{output_path}/{file_index}.csv', 'w', newline='')
                writer = csv.writer(output_file)
                writer.writerow(header)  # 写入头部
                file_index += 1
                row_count = 0

            writer.writerow(row)
            row_count += 1

        if output_file is not None:
            output_file.close()

# 示例用法
split_csv('input.csv', 'output', 1000)

上述代码中,split_csv函数接受三个参数:输入文件路径、输出文件夹路径和每个输出文件的行数限制。函数使用CSV模块读取输入文件,并逐行处理。

在处理行数达到限制或需要新的输出文件时,函数会关闭当前的输出文件(如果有),并创建一个新的输出文件。在每个输出文件中,函数会先写入头部,然后逐行写入行数据。

在示例中,split_csv('input.csv', 'output', 1000)将会读取名为input.csv的输入文件,将其拆分成每个文件最多包含1000行的输出文件,并保存到名为output的文件夹中。

相关内容

热门资讯

wepoke ai辅助!wep... wepoke ai辅助!wepoke可以使用模拟器,wepok软件透明挂,攻略教程(有挂技巧)1、点...
wepoke辅助挂!wepok... wepoke辅助挂!wepoke有插件,wepOkE总是真的有挂,科技教程(有挂细节);玩家必备必赢...
玩家攻略推荐!天天斗牌大联盟麻... 玩家攻略推荐!天天斗牌大联盟麻将(透明挂)好像真的有挂(2021已更新)(哔哩哔哩)1、构建自己的天...
微扑克有辅助挂!微扑克大厅都是... 微扑克有辅助挂!微扑克大厅都是机器人,德州扑克微扑克俱乐部,系统教程(有挂机密)是一款可以让一直输的...
wepokeai机器人!wep... 这是一款非常优秀的WepOke ia辅助检测软件,能够让你了解到WepOke中牌率当中全部隐藏参数,...
揭秘一下!科乐麻将系统规律(透... 揭秘一下!科乐麻将系统规律(透视)原来是有挂(2026已更新)(哔哩哔哩)1、科乐麻将系统规律系统规...
微扑克有辅助挂!微扑克有后台控... 微扑克有辅助挂!微扑克有后台控制(透明挂)原来真的是有挂1、超多福利:超高返利,海量正版游戏,微扑克...
WePoKe外 挂!wopok... 1、WePoKe外 挂!wopoker有外 挂(透明挂)wEpOke(就是真的有挂);该软件可以轻松...
程序员教你!欢乐划水麻将是不是... 程序员教你!欢乐划水麻将是不是有猫腻(透视辅助)都是有挂(2024已更新)(哔哩哔哩)1、点击下载安...
微扑克系统发牌规律!微扑克有计... 1、微扑克系统发牌规律!微扑克有计算器,微扑克ai软件(确实真的有挂);代表性(透视辅助软件透明挂)...