使用 Python 读取并写入 CSV 文件

Document 对象参考手册 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)

代码解析:

  1. import csv:导入 Python 的 csv 模块,该模块提供了读取和写入 CSV 文件的功能。
  2. with open('input.csv', mode='r', newline='', encoding='utf-8') as infile::以只读模式打开名为 input.csv 的文件,并指定编码为 utf-8newline='' 是为了确保在不同操作系统上都能正确处理换行符。
  3. reader = csv.reader(infile):创建一个 CSV 读取器对象,用于逐行读取文件内容。
  4. data = [row for row in reader]:使用列表推导式将文件中的每一行数据存储到 data 列表中。
  5. with open('output.csv', mode='w', newline='', encoding='utf-8') as outfile::以写入模式打开名为 output.csv 的文件,并指定编码为 utf-8
  6. writer = csv.writer(outfile):创建一个 CSV 写入器对象,用于将数据写入文件。
  7. 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

Document 对象参考手册 Python3 实例