用Python读取和写入CSV文件的方法
CSV(Comma Separated Values)文件是一种以逗号分隔字段的文本文件格式,通常用于存储表格数据。Python内置了一个csv模块,方便读取和写入CSV文件。
首先,我们来看如何读取CSV文件。
读取CSV文件的步骤如下:
1. 引入csv模块:import csv
2. 打开CSV文件:使用内置的open()函数以读模式打开CSV文件,并将文件对象赋值给一个变量,比如file。
file = open('example.csv', 'r')
3. 创建一个CSV读取器:使用csv.reader()函数创建一个CSV读取器对象并将文件对象作为参数传递给它。
reader = csv.reader(file)
4. 读取CSV文件内容:使用for循环遍历读取器对象,每次迭代获取一行CSV数据,以列表形式返回。
for row in reader:
print(row)
完整的读取CSV文件的示例代码如下:
import csv
file = open('example.csv', 'r')
reader = csv.reader(file)
for row in reader:
print(row)
file.close()
接下来,我们来看一下如何写入CSV文件。
写入CSV文件的步骤如下:
1. 引入csv模块:import csv
2. 打开/创建CSV文件:使用内置的open()函数以写模式打开或创建CSV文件,并将文件对象赋值给一个变量,比如file。
file = open('example.csv', 'w', newline='')
3. 创建一个CSV写入器:使用csv.writer()函数创建一个CSV写入器对象并将文件对象作为参数传递给它。
writer = csv.writer(file)
4. 写入CSV文件内容:使用写入器对象的writerow()方法将每行数据写入CSV文件。
writer.writerow(['Name', 'Age', 'Country']) writer.writerow(['John Doe', '25', 'United States']) writer.writerow(['Jane Smith', '30', 'Canada'])
5. 关闭CSV文件:使用close()方法关闭CSV文件。
file.close()
完整的写入CSV文件的示例代码如下:
import csv
file = open('example.csv', 'w', newline='')
writer = csv.writer(file)
writer.writerow(['Name', 'Age', 'Country'])
writer.writerow(['John Doe', '25', 'United States'])
writer.writerow(['Jane Smith', '30', 'Canada'])
file.close()
注意事项:
- 使用open()函数打开CSV文件时,第二个参数指定文件模式,'r'表示只读,'w'表示写(会覆盖原有内容),'a'表示追加。
- 创建CSV写入器对象时,可以使用delimiter参数指定字段分隔符,默认为逗号。
- 创建CSV写入器对象时,可以使用quotechar参数指定字段引用字符,默认为双引号。
这就是使用Python读取和写入CSV文件的方法,希望对你有帮助!
