使用 Python 读取并写入 CSV 文件

使用 Python 读取并写入 CSV 文件

Python3 实例

在 Python 中，我们可以使用内置的 csv 模块来读取和写入 CSV 文件。CSV 文件是一种常见的文件格式，用于存储表格数据。下面是一个简单的示例，展示如何读取一个 CSV 文件并将其内容写入另一个 CSV 文件。

实例

import csv

# 读取 CSV 文件

with open('input.csv', mode='r', newline='', encoding='utf-8') as infile:

reader = csv.reader(infile)

data = [row for row in reader]

# 写入 CSV 文件

with open('output.csv', mode='w', newline='', encoding='utf-8') as outfile:

writer = csv.writer(outfile)

writer.writerows(data)

代码解析：

import csv：导入 Python 的 csv 模块，该模块提供了读取和写入 CSV 文件的功能。

with open('input.csv', mode='r', newline='', encoding='utf-8') as infile:：以只读模式打开名为 input.csv 的文件，并指定编码为 utf-8。newline='' 是为了确保在不同操作系统上都能正确处理换行符。

reader = csv.reader(infile)：创建一个 CSV 读取器对象，用于逐行读取文件内容。

data = [row for row in reader]：使用列表推导式将文件中的每一行数据存储到 data 列表中。

with open('output.csv', mode='w', newline='', encoding='utf-8') as outfile:：以写入模式打开名为 output.csv 的文件，并指定编码为 utf-8。

writer = csv.writer(outfile)：创建一个 CSV 写入器对象，用于将数据写入文件。

writer.writerows(data)：将 data 列表中的所有行写入到 output.csv 文件中。

输出结果：

假设 input.csv 文件内容如下：

Name,Age,City

Alice,30,New York

Bob,25,Los Angeles

Charlie,35,Chicago

运行上述代码后，output.csv 文件将包含与 input.csv 相同的内容：

Name,Age,City

Alice,30,New York

Bob,25,Los Angeles

Charlie,35,Chicago

Python3 实例