使用Python的CSV函数库读取和写入CSV文件
在Python中,可以使用csv库来读取和写入CSV文件。CSV(逗号分隔值)是一种常用的文件格式,它以逗号作为分隔符来存储表格数据。CSV文件通常用于数据交换和数据存储,因为它可以被各种程序和工具轻松地读取和处理。
首先,我们需要导入csv库:
import csv
读取CSV文件
一旦导入了csv库,我们就可以使用其提供的函数来读取CSV文件。最常用的函数是reader(),它可以读取CSV文件中的数据并返回一个列表。以下是读取CSV文件的基本方法:
# 打开CSV文件
with open('data.csv', 'r') as file:
# 创建CSV读取器
csv_reader = csv.reader(file)
# 逐行读取CSV数据
for row in csv_reader:
# 使用row变量来访问每一行的数据
print(row)
以上代码中,我们首先使用open()函数打开一个名为data.csv的CSV文件,并指定'r'作为文件的模式,表示读取文件。然后,我们使用csv.reader()函数创建一个CSV读取器,它将文件对象作为参数。最后,我们使用for循环逐行读取CSV数据,每一行数据都被存储在名为row的变量中,我们可以对其进行处理。
写入CSV文件
与读取CSV文件类似,我们可以使用csv库中的writer()函数来写入CSV文件。以下是写入CSV文件的基本方法:
# 创建CSV数据
data = [
['Name', 'Age', 'Gender'],
['John Smith', '25', 'Male'],
['Jane Doe', '30', 'Female'],
['Mike Johnson', '35', 'Male']
]
# 打开CSV文件
with open('data.csv', 'w', newline='') as file:
# 创建CSV写入器
csv_writer = csv.writer(file)
# 写入CSV数据
for row in data:
csv_writer.writerow(row)
在以上代码中,我们首先创建了一个名为data的列表,其中包含了要写入CSV文件的数据。然后,我们使用open()函数打开一个名为data.csv的CSV文件,并指定'w'作为文件的模式,表示写入文件。接下来,我们使用csv.writer()函数创建一个CSV写入器,它将文件对象作为参数。最后,我们使用for循环逐行写入CSV数据,每一行数据被存储在名为row的变量中,我们通过调用csv_writer对象的writerow()函数将其写入CSV文件中。
总结
使用Python的csv库可以轻松地读取和写入CSV文件。通过使用csv.reader()函数可以逐行读取CSV文件中的数据,并通过使用csv.writer()函数可以逐行写入CSV文件中的数据。
