Python如何读取、写入CSV数据

Python如何读取、写入CSV数据

1. 了解CSV文件

CSV（Comma Separated Values）是一种常见的文件格式，用于存储表格数据。它由逗号分隔的字段组成，每行表示一条记录。在Python中读取和写入CSV文件非常方便，我们可以使用内置的csv模块来处理。

2. 读取CSV文件

要读取CSV文件，我们首先需要导入csv模块，并使用内置的open函数打开文件。接下来，我们可以使用csv.reader函数创建一个reader对象，并通过遍历此对象来逐行读取文件内容。

import csv
# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建reader对象
    reader = csv.reader(file)
    
    # 遍历每一行
    for row in reader:
        # 处理每一行的数据
        print(row)

在上述示例中，我们使用open函数打开名为"data.csv"的CSV文件，并将文件对象赋给变量file。然后，我们使用csv.reader函数创建名为reader的reader对象。通过遍历reader对象，我们可以逐行打印文件的内容。

3. 写入CSV文件

要写入CSV文件，我们同样需要导入csv模块，并使用内置的open函数打开文件。接下来，我们可以使用csv.writer函数创建一个writer对象，并使用writerow方法逐行写入数据。

import csv
# 准备要写入的数据
data = [
    ['Name', 'Age', 'Country'],
    ['John', '25', 'USA'],
    ['Emily', '30', 'Canada'],
    ['Michael', '35', 'UK']
]
# 打开CSV文件
with open('output.csv', 'w') as file:
    # 创建writer对象
    writer = csv.writer(file)
    
    # 写入数据
    writer.writerows(data)

在上述示例中，我们准备了一个数据列表data，其中包含了要写入的数据。使用open函数打开名为"output.csv"的CSV文件并将文件对象赋给变量file。然后，我们使用csv.writer函数创建writer对象，并使用writerow方法将数据逐行写入文件。

总结

通过使用csv模块，Python可以轻松地读取和写入CSV文件。使用csv.reader函数和csv.writer函数可以分别创建reader对象和writer对象，然后通过遍历reader对象和使用writerow方法来逐行读取和写入文件数据。

CSV文件是一种非常常见的数据存储格式，在数据处理和数据分析中经常会使用到。掌握在Python中读取和写入CSV文件的方法，对于数据处理工作是非常有帮助的。