Python逐行写CSV文件

什么是CSV文件？

CSV（Comma-Separated Values）文件是一种常用的纯文本文件格式，用于存储表格数据。每行代表一个表格的记录，每个字段用逗号或其他分隔符进行分割。CSV文件可以使用电子表格软件进行编辑和查看，也可以作为程序之间交换数据的一种通用格式。

Python中的CSV模块

Python提供了csv模块来读写CSV文件。csv模块提供了一组函数和类，可以方便地处理CSV文件。

写入CSV文件

通常情况下，我们可以一次性将数据写入CSV文件。但有时候，我们可能需要逐行写入数据，比如在处理大量数据时，逐行写入可以减少内存的占用。

下面是一个逐行写入CSV文件的例子：

import csv

# 创建CSV文件
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)

    # 写入表头
    writer.writerow(['Name', 'Age', 'Email'])

    # 写入数据行
    writer.writerow(['Alice', 25, 'alice@example.com'])
    writer.writerow(['Bob', 30, 'bob@example.com'])
    writer.writerow(['Charlie', 35, 'charlie@example.com'])

首先，我们使用open函数创建一个CSV文件，并指定写入模式和newline=''参数。newline=''参数是为了防止在Windows系统中出现换行问题。

接下来，我们创建一个csv.writer对象，并将文件对象传递给它。然后，我们可以使用writerow方法逐行写入数据。在上面的例子中，我们首先写入了表头，然后逐行写入了数据。

CSV文件的读取和写入

另一种常见的情况是从一个CSV文件读取数据，并将处理后的数据写入另一个CSV文件。下面是一个例子：

import csv

# 读取原始CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    next(reader)  # 跳过表头

    # 处理数据
    data = []
    for row in reader:
        name = row[0]
        age = int(row[1])
        email = row[2].strip()
        data.append([name, age, email])

# 写入新的CSV文件
with open('processed_data.csv', 'w', newline='') as file:
    writer = csv.writer(file)

    # 写入表头
    writer.writerow(['Name', 'Age', 'Email'])

    # 写入处理后的数据
    writer.writerows(data)

上面的例子首先使用csv.reader对象读取原始CSV文件，并通过next函数跳过了表头。然后，我们对每一行数据进行处理，取出姓名、年龄和电子邮件，并将处理后的数据存储在一个列表中。

最后，我们创建一个新的CSV文件，并使用csv.writer对象将处理后的数据逐行写入文件。

总结

本文介绍了如何使用Python逐行写入CSV文件。我们可以使用csv.writer对象的writerow方法逐行写入数据，也可以使用writerows方法一次性写入多行数据。在处理大量数据时，逐行写入可以降低内存的占用。同时，我们还学习了如何从CSV文件中读取数据并写入到另一个CSV文件中。

CSV文件是一种常用的数据交换格式，Python的csv模块提供了一组方便的函数和类来处理CSV文件。掌握如何逐行写入CSV文件，将有助于我们更好地处理和分析数据。

关系图

下面是本文中代码的关系图示例：

erDiagram
    CSV文件 --|> 写入CSV文件
    CSV文件 --|> 读取和写入
    读取和写入 --|> 逐行写入

以上就是Python逐行写入CSV文件的介绍。希望本文对你有所帮助！