python csv 写文件头

原创

mob649e8166858d 2023-10-19 07:15:57 ©著作权

文章标签 CSV 写入文件数据 文章分类 Python 后端开发

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

Python CSV 写文件头

在处理 CSV（Comma-Separated Values，逗号分隔值）文件时，我们经常需要在文件的第一行写入文件头。文件头是指文件中每一列的名称，它提供了对数据的解释和说明，使得数据更易于理解和处理。

本文将介绍如何使用 Python 的 csv 模块来写入 CSV 文件的文件头，并提供一些代码示例。

csv 模块简介

csv 模块是 Python 内置的用于处理 CSV 文件的模块，它提供了一组方法和类来读取和写入 CSV 文件。使用 csv 模块，我们可以方便地操作 CSV 文件，如读取数据、写入数据、添加文件头等。

写入文件头

要写入 CSV 文件的文件头，我们可以使用 csv 模块中的 writerow() 方法。该方法接受一个可迭代对象作为参数，将其写入 CSV 文件的一行。

下面是一个示例代码，演示了如何写入 CSV 文件的文件头：

import csv

header = ['Name', 'Age', 'City']

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

在上面的代码中，我们首先定义了一个列表 header，其中包含了文件头的内容。然后，使用 open() 函数打开文件，并指定文件打开模式为 'w'（写入模式）。newline='' 参数用于控制换行符的处理方式。

接下来，我们创建了一个 csv.writer 对象，并将文件对象传递给它。然后，调用 writerow() 方法，传入文件头列表 header，将其写入到 CSV 文件中。

写入文件头后，我们可以继续写入数据行。例如，我们可以再次调用 writerow() 方法来写入一行数据：

data = ['John Doe', 25, 'New York']
writer.writerow(data)

完整示例

下面是一个完整的示例代码，演示了如何使用 csv 模块写入 CSV 文件的文件头和数据行：

import csv

header = ['Name', 'Age', 'City']
data = [
    ['John Doe', 25, 'New York'],
    ['Jane Smith', 30, 'San Francisco'],
    ['Tom Johnson', 35, 'Chicago']
]

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

在上面的代码中，我们定义了一个二维列表 data，其中包含了三行数据。然后，使用 open() 函数创建文件对象，并打开文件。接下来，创建 csv.writer 对象，并在每行写入数据之前先写入文件头，最后调用 writerows() 方法写入所有数据行。