使用 Python 读取并写入 CSV 文件
在 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)
# 读取 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
点我分享笔记