如何使用Python函数对CSV文件进行读写操作?
发布时间:2023-06-01 01:42:47
CSV(Comma-Separated Values)是一种轻量级的文本格式,通常用于表格数据存储和传输。在Python中,需要对CSV文件进行读写操作时,可以使用csv模块提供的函数来实现。
1.读取CSV文件
要读取CSV文件,需要使用csv模块中的reader函数。该函数接受一个文件对象作为参数,并将文件解析为一行一行的数据列表。每行数据列表中的元素,就是以逗号分隔的各个字段。
可以使用以下代码读取CSV文件:
import csv
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
上述代码中,首先使用with open语句打开名为data.csv的文件,并指定读取模式。然后使用csv.reader函数读取文件对象,并将其存储到变量reader中。接着使用for循环遍历reader对象,依次输出每行数据列表。
2.写入CSV文件
要将数据写入CSV文件,可以使用csv模块中的writer函数。该函数接受一个文件对象和一个可迭代对象作为参数,通过迭代对象将数据写入到CSV文件中。
可以使用以下代码将数据写入CSV文件:
import csv
data = [
['Name', 'Age', 'City'],
['Alice', '22', 'New York'],
['Bob', '30', 'Chicago'],
['Charlie', '28', 'San Francisco']
]
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
for row in data:
writer.writerow(row)
上述代码中,首先定义一个二维列表data,表示要写入的数据。然后使用with open语句打开名为output.csv的文件,并指定写入模式。注意指定newline=''参数,以避免在Windows系统中出现多余的空行。然后使用csv.writer函数创建一个写入对象writer,并将其存储到变量writer中。接着使用for循环遍历data列表,依次将每行数据写入CSV文件中。
以上是使用Python函数对CSV文件进行读写操作的基本方法,当然,csv模块还提供了许多其他函数和参数来更有效地操作CSV文件。建议在使用时先仔细查看模块文档和函数说明。
