Python 修改CSV文件实例详解

1. CSV文件的基本介绍

CSV（Comma-Separated Values）是一种常用的文本文件格式，用来存储结构化的数据。它由多行数据记录组成，每行记录包含一个或多个字段，字段之间使用逗号进行分隔。

CSV文件通常用于数据交换和数据存储，由于其简单、易读、易写的特性，得到了广泛的应用。在Python中，可以使用标准库中的csv模块进行CSV文件的读写操作。

2. 读取CSV文件

2.1 使用csv.reader()函数读取CSV文件

使用csv.reader()函数可以按行读取CSV文件，并将每一行数据解析为一个列表。以下是读取CSV文件的示例代码：

import csv

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

由于逗号是CSV文件的默认分隔符，所以在默认情况下不需要显式地指定分隔符。读取的每一行数据都会被解析为一个列表，可以通过索引访问每个字段的值。

2.2 使用pandas库读取CSV文件

pandas是Python中一种基于NumPy的数据处理库，提供了高效、灵活的数据结构和数据分析工具。pandas库中的read_csv()函数可以方便地读取CSV文件，并将其解析为DataFrame对象。

以下是使用pandas库读取CSV文件的示例代码：

import pandas as pd

# 读取CSV文件
data = pd.read_csv('data.csv')
# 打印DataFrame对象
print(data)

通过read_csv()函数读取的数据将会被解析为一个DataFrame对象，DataFrame是pandas库中最常用的数据结构，类似于电子表格或SQL中的数据表。

3. 修改CSV文件

如果想要修改CSV文件中的数据，可以直接通过修改DataFrame对象或利用csv.writer()函数来实现。

3.1 修改DataFrame对象

使用pandas库读取CSV文件后，得到的是一个DataFrame对象，可以通过对DataFrame对象的操作来修改数据。以下是修改DataFrame对象的示例代码：

import pandas as pd

# 读取CSV文件
data = pd.read_csv('data.csv')
# 修改数据
data['temperature'] = data['temperature'] * 0.6
# 保存修改后的数据到CSV文件
data.to_csv('data_modified.csv', index=False)

上述代码中，先读取CSV文件并将数据存储到一个DataFrame对象中，然后通过修改DataFrame的某一列来修改相应的数据。最后，将修改后的数据保存到CSV文件中，注意设置index参数为False，避免写入的CSV文件中包含索引列。

3.2 使用csv.writer()函数修改CSV文件

除了使用pandas库外，还可以使用csv.writer()函数来直接修改CSV文件。csv.writer()函数接受一个文件对象，并提供了写入CSV文件的方法。

以下是使用csv.writer()函数修改CSV文件的示例代码：

import csv

# 打开原始CSV文件和新的CSV文件
with open('data.csv', 'r') as file1, open('data_modified.csv', 'w', newline='') as file2:
    # 创建reader对象和writer对象
    reader = csv.reader(file1)
    writer = csv.writer(file2)
    
    # 遍历每一行数据
    for row in reader:
        # 修改数据
        row[1] = float(row[1]) * 0.6
        
        # 写入新的CSV文件
        writer.writerow(row)

上述代码中，首先打开原始的CSV文件和一个新的CSV文件，然后创建csv.reader对象和csv.writer对象，通过遍历原始CSV文件中的每一行数据，修改数据后写入新的CSV文件。

总结

本文介绍了如何使用Python来读取和修改CSV文件的方法。通过Python的csv模块或pandas库，可以方便地进行CSV文件的读写操作。在读取CSV文件时，可以使用csv.reader()函数按行读取或使用pandas库的read_csv()函数将CSV文件解析为DataFrame对象。在修改CSV文件时，可以通过修改DataFrame对象或利用csv.writer()函数来实现。希望本文能对你理解和使用Python来修改CSV文件有所帮助。