欢迎访问宙启技术站
智能推送

用Python读取和写入CSV文件的方法

发布时间:2023-12-26 23:15:22

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文件的方法,希望对你有帮助!