按行拆分CSV文件
创始人
2024-08-22 01:30:20
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是每个输出文件的行数。

相关内容

热门资讯

安装apache-beam==... 出现此错误可能是因为用户的Python版本太低,而apache-beam==2.34.0需要更高的P...
避免在粘贴双引号时向VS 20... 在粘贴双引号时向VS 2022添加反斜杠的问题通常是由于编辑器的自动转义功能引起的。为了避免这个问题...
Android Recycle... 要在Android RecyclerView中实现滑动卡片效果,可以按照以下步骤进行操作:首先,在项...
omi系统和安卓系统哪个好,揭... OMI系统和安卓系统哪个好?这个问题就像是在问“苹果和橘子哪个更甜”,每个人都有自己的答案。今天,我...
原生ios和安卓系统,原生对比... 亲爱的读者们,你是否曾好奇过,为什么你的iPhone和安卓手机在操作体验上有着天壤之别?今天,就让我...
Android - 无法确定任... 这个错误通常发生在Android项目中,表示编译Debug版本的Java代码时出现了依赖关系问题。下...
Android - NDK 预... 在Android NDK的构建过程中,LOCAL_SRC_FILES只能包含一个项目。如果需要在ND...
Akka生成Actor问题 在Akka框架中,可以使用ActorSystem对象生成Actor。但是,当我们在Actor类中尝试...
Agora-RTC-React... 出现这个错误原因是因为在 React 组件中使用,import AgoraRTC from “ago...
Alertmanager在pr... 首先,在Prometheus配置文件中,确保Alertmanager URL已正确配置。例如:ale...