如何使用Python中的csv库中的函数来读写CSV文件?
Python中的csv库提供了一组函数,用于读取和写入CSV文件。CSV是一种常见的文件格式,用于存储表格数据,每行表示一条记录,通过逗号或其他分隔符来分隔不同的字段。
要使用csv库,首先需要导入它:
import csv
接下来,我们将介绍一些常用的csv库函数和它们的用法。
1. 读取CSV文件
使用csv库的reader()函数可以方便地读取CSV文件中的数据。它可以接受一个文件对象作为参数,并返回一个可迭代的reader对象,可以用于遍历CSV文件中的每一行数据。
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
在上面的例子中,data.csv是CSV文件的文件名。使用open()函数打开文件,并将文件对象作为参数传递给reader()函数。然后使用for循环遍历reader对象,可以逐行打印CSV文件中的数据。
2. 写入CSV文件
csv库中的writer()函数可以用于将数据写入到CSV文件中。它也接受一个文件对象作为参数,并返回一个writer对象,可以用于将数据写入CSV文件。
data = [
['Name', 'Age', 'Gender'],
['John', '25', 'Male'],
['Emma', '28', 'Female'],
['Michael', '30', 'Male']
]
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerows(data)
在上面的例子中,data是一个包含要写入CSV文件中的数据的列表。使用open()函数并传递'w'作为打开文件的模式,newline=''参数用于避免写入额外的空行。然后使用writer()函数创建一个writer对象,并使用writerows()函数将数据写入CSV文件。
3. 设置分隔符
默认情况下,csv.writer对象使用逗号作为字段的分隔符。但是,你也可以通过设置delimiter参数来使用其他分隔符。
data = [
['Name', 'Age', 'Gender'],
['John', '25', 'Male'],
['Emma', '28', 'Female'],
['Michael', '30', 'Male']
]
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file, delimiter=';')
writer.writerows(data)
在上面的例子中,我们将分隔符设置为分号(;),并使用writer()函数创建writer对象。
4. 指定CSV文件的其他参数
csv.writer和csv.reader函数还可以接受其他参数,以指定如何处理引号、行结束符等。例如,quotechar参数可以设置引号字符,lineterminator参数可以设置行结束符。
data = [
['Name', 'Age', 'Gender'],
['John', '25', 'Male'],
['Emma', '28', 'Female'],
['Michael', '30', 'Male']
]
with open('output.csv', 'w', newline='') as file:
writer = csv.writer(file, quotechar='"', lineterminator='
\r')
writer.writerows(data)
在上面的例子中,我们将引号字符设置为双引号("),行结束符设置为
\r。
以上就是使用Python中的csv库读写CSV文件的基本方法。通过这些函数,我们可以方便地读取CSV文件中的数据,并将数据写入CSV文件。
