使用 Python 读写 CSV 文件

CSV（逗号分隔值）文件广泛用于数据存储和交换。在 Python 中，您可以使用内置的 csv 模块轻松读取和写入 CSV 文件。本文将指导您了解在 Python 中使用 CSV 文件的基础知识。

读取 CSV 文件

要读取 CSV 文件，您可以使用 csv 模块中的 csv.reader 类。这是一个简单示例：

import csv

# Open the CSV file
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    
    # Iterate over each row in the CSV
    for row in reader:
        print(row)

在此示例中，example.csv 以读取模式打开，并使用 csv.reader 解析文件。CSV 文件的每一行都以值列表的形式打印。

写入 CSV 文件

要将数据写入 CSV 文件，请使用 csv.writer 类。以下是将行列表写入 CSV 文件的方法：

import csv

# Data to be written to the CSV
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles'],
    ['Charlie', 35, 'Chicago']
]

# Open the CSV file in write mode
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # Write each row to the CSV file
    writer.writerows(data)

在此示例中，output.csv 被创建或用指定数据覆盖。writer.writerows() 方法用于一次写入多行。

结论

使用 csv 模块，在 Python 中处理 CSV 文件非常简单。您可以轻松读取和写入 CSV 文件，从而轻松处理这种常见格式的数据。