Python 2.x 中如何使用csv模块读取和写入CSV文件
导言:
CSV(Comma Separated Values)是一种常见的文件格式,用于存储和交换数据。Python的csv模块提供了一种简单的方式来读取和写入CSV文件。本文将介绍如何使用csv模块在Python 2.x中读取和写入CSV文件,并提供相应的代码示例。
一、读取CSV文件
要读取CSV文件,首先需要导入csv模块。然后,通过打开文件并创建一个csv.reader对象来读取CSV文件的内容。
代码示例:
import csv # 打开CSV文件 with open('data.csv', 'r') as file: # 创建csv.reader对象 reader = csv.reader(file) # 读取文件内容 for row in reader: # 打印每一行数据 print(row)
在上述代码中,我们使用with open('data.csv', 'r') as file
打开CSV文件,并创建了一个csv.reader对象。然后,我们可以通过遍历csv.reader对象来逐行读取文件的内容。每一行的数据以列表的形式返回,我们可以对每一行进行相应的处理。
二、写入CSV文件
要将数据写入CSV文件,同样需要先导入csv模块。然后,通过打开文件并创建一个csv.writer对象来写入CSV文件。
代码示例:
import csv # 要写入的数据 data = [ ['Name', 'Age', 'Gender'], ['Alice', '20', 'Female'], ['Bob', '25', 'Male'], ['Charlie', '30', 'Male'] ] # 打开CSV文件 with open('data.csv', 'w') as file: # 创建csv.writer对象 writer = csv.writer(file) # 写入数据 for row in data: writer.writerow(row)
在上述代码中,我们首先定义了要写入CSV文件的数据,这里使用一个列表表示。然后,我们使用with open('data.csv', 'w') as file
打开CSV文件,并创建了一个csv.writer对象。然后,我们通过遍历数据,并使用writer.writerow(row)
方法将每一行数据写入文件。
三、指定分隔符
默认情况下,csv模块使用逗号作为分隔符。如果CSV文件使用其他字符作为分隔符,可以通过指定delimiter
参数来进行设置。
代码示例:
import csv # 打开CSV文件,指定分隔符为分号 with open('data.csv', 'r') as file: reader = csv.reader(file, delimiter=';') for row in reader: print(row)
在上述代码中,我们使用delimiter
参数将分隔符设置为分号。这样,读取CSV文件时,csv.reader将使用分号作为每一行数据的分隔符。
总结:
本文介绍了如何使用Python 2.x中的csv模块读取和写入CSV文件,并提供了相应的代码示例。读取CSV文件时,需要创建一个csv.reader对象,并通过遍历该对象来逐行读取文件的内容。写入CSV文件时,需要创建一个csv.writer对象,并使用writer.writerow(row)
方法将每一行数据写入文件。另外,我们还介绍了如何通过指定delimiter
参数来设置分隔符。以上方法可以帮助你使用Python 2.x处理CSV文件中的数据。