python csv首行固定

原创

mob649e81607bf3 2023-08-02 14:01:48 ©著作权

文章标签 CSV 数据 python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e81607bf3的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python CSV首行固定实现教程

简介

在处理CSV文件时，有时我们需要保持CSV文件的首行不变，即使对文件进行了其他操作，例如排序、过滤、添加数据等。本文将教你如何使用Python实现CSV文件首行固定的功能。

流程图

下面是一个展示整个流程的简单表格：

步骤	操作
1	打开原始CSV文件
2	读取首行数据
3	执行其他操作
4	写入首行数据到新的CSV文件
5	写入其他操作结果到新的CSV文件
6	关闭文件

代码实现

步骤1：打开原始CSV文件

import csv

with open('original.csv', 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

首先，我们使用open函数以只读模式打开原始的CSV文件，然后使用csv.reader函数创建一个reader对象，将文件内容读取为一个二维数组（每一行是一个列表）。最后，我们将读取的数据保存在一个名为data的变量中。

步骤2：读取首行数据

header = data[0]

在我们的例子中，首行数据（即表头）保存在data变量的第一个元素中。我们将该数据保存在名为header的变量中。

步骤3：执行其他操作

在这一步中，你可以对CSV文件执行任何你想要的操作，例如排序、过滤、添加数据等。这部分的代码将根据你的具体需求而定，并不在本文的范围之内。

步骤4：写入首行数据到新的CSV文件

with open('new.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(header)

我们使用open函数以写入模式打开一个新的CSV文件，并使用csv.writer函数创建一个writer对象。然后，我们使用writerow方法将保存在header变量中的首行数据写入到新的CSV文件中。请注意，我们在打开文件时传入了newline=''参数，这是为了避免在Windows系统中产生额外的空行。

步骤5：写入其他操作结果到新的CSV文件

with open('new.csv', 'a', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data[1:])

在这一步中，我们将执行其他操作后的结果写入到新的CSV文件中。同样地，我们使用open函数以追加模式打开刚才创建的新的CSV文件，并使用csv.writer函数创建一个writer对象。然后，我们使用writerows方法将从第二行开始的数据写入到新的CSV文件中。注意，我们使用了切片操作符[1:]来排除首行数据。