按行拆分CSV文件
创始人
2024-08-22 01:30:13
0

以下是一个示例代码,将CSV文件按行拆分成多个文件:

import csv

def split_csv_file(input_file, output_prefix, chunk_size):
    with open(input_file, 'r') as file:
        reader = csv.reader(file)

        header = next(reader)  # 读取并保存CSV文件的头部

        file_index = 1
        row_count = 0

        output_file = f"{output_prefix}_{file_index}.csv"  # 输出文件的文件名

        with open(output_file, 'w', newline='') as output:
            writer = csv.writer(output)

            writer.writerow(header)  # 将头部写入输出文件

            for row in reader:
                writer.writerow(row)  # 将行写入输出文件

                row_count += 1

                if row_count >= chunk_size:
                    file_index += 1
                    row_count = 0

                    output_file = f"{output_prefix}_{file_index}.csv"

                    output.close()  # 关闭当前输出文件
                    output = open(output_file, 'w', newline='')  # 打开新的输出文件
                    writer = csv.writer(output)

            output.close()

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

这个示例代码使用Python的csv模块来读取和写入CSV文件。它首先读取输入文件的头部,并将其写入所有输出文件。然后,它逐行读取输入文件,并将每一行写入当前输出文件。当达到指定的行数(chunk_size)时,会关闭当前输出文件,打开一个新的输出文件,并继续写入剩余的行。

在示例用法中,input.csv是要拆分的输入文件,output是输出文件的前缀(例如,如果outputoutput,则生成的文件名将为output_1.csvoutput_2.csv等),1000是每个输出文件的行数。

相关内容

热门资讯

安装Pillow时遇到了问题:... 遇到这个问题,可能是因为缺少libwebpmux3软件包。解决方法是手动安装libwebpmux3软...
安装Python库"... 安装Python库"firedrake"的解决方法如下:打开终端或命令提示符(Windows系统)。...
安装ug未能链接到许可证服务器 安装UG未能链接到许可证服务器是UG用户在安装软件时常遇到的问题之一。该问题的解决方法需要技术向的知...
安装React Native时... 当安装React Native时出现构建错误的情况,可以尝试以下解决方法:确保已经安装了最新版本的C...
按转换模式过滤日志【%t】。 要按照转换模式过滤日志,可以使用正则表达式来实现。下面是一个示例代码,使用Java语言的Patter...
安装react-native-... 要安装react-native-onesignal并在应用关闭时仍能接收通知,可以按照以下步骤进行:...
安装Rails时构建webso... 在安装Rails时,如果构建websocket-driver时发生错误,可以尝试以下解决方法:更新系...
Apache Nifi在Kub... Apache Nifi可以在Kubernetes上运行,并且已经准备好用于生产环境。下面是一个使用H...
安卓 - 谷歌地图卡住了 问题描述:在安卓设备上使用谷歌地图应用时,地图卡住了,无法进行任何操作。解决方法一:清除应用缓存和数...
安装React-Scripts... 这是因为React-Scripts使用Facebook工具包中的一些脚本。 joinAdIntere...