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

如何使用Python读取和写入CSV文件

发布时间:2023-12-04 02:34:29

CSV(逗号分隔值)是一种常用的文本文件格式,用于存储数据表的信息。Python提供了多种处理CSV文件的方式,可以通过csv模块实现读取和写入操作。

对于读取CSV文件,步骤如下:

1. 导入csv模块,import csv

2. 打开CSV文件,使用open()函数,指定文件名和访问模式(通常为'r')。

3. 创建一个csv.reader对象,将打开的文件对象作为参数传入,reader = csv.reader(file_obj)

4. 使用next()函数,读取CSV文件的第一行,通常为表头,header = next(reader)

5. 使用循环遍历reader对象,读取每一行的数据:

- 使用for循环,for row in reader

- 每一行的数据是一个列表,可以通过索引或循环来提取每个元素。

6. 关闭文件,使用file_obj.close()

示例代码如下:

import csv

# 打开CSV文件
with open('data.csv', 'r') as file_obj:
    # 创建reader对象
    reader = csv.reader(file_obj)
    
    # 读取表头
    header = next(reader)
    
    # 读取每一行数据
    for row in reader:
        print(row)
    
    # 关闭文件
    file_obj.close()

对于写入CSV文件,步骤如下:

1. 导入csv模块,import csv

2. 打开CSV文件,使用open()函数,指定文件名和访问模式(通常为'w')。

3. 创建一个csv.writer对象,将打开的文件对象作为参数传入,writer = csv.writer(file_obj)

4. 使用writerow()方法,写入每一行的数据:

- 以列表的形式传入数据,writer.writerow(['data1', 'data2', 'data3'])

- 可以通过循环的方式写入多行数据。

5. 关闭文件,使用file_obj.close()

示例代码如下:

import csv

# 打开CSV文件
with open('data.csv', 'w') as file_obj:
    # 创建writer对象
    writer = csv.writer(file_obj)
    
    # 写入表头
    writer.writerow(['Header1', 'Header2', 'Header3'])
    
    # 写入数据
    writer.writerow(['data1', 'data2', 'data3'])
    writer.writerow(['data4', 'data5', 'data6'])
    
    # 关闭文件
    file_obj.close()

通过以上步骤,可以使用Python读取和写入CSV文件,实现对数据的读取和存储。其中要注意文件的打开和关闭操作,确保资源的正确释放。此外,还可以使用csv.DictReadercsv.DictWriter来处理带有表头的CSV文件,以字典的形式读写数据,更加方便直观。