python 下CSV 操作-猿码集

Python下CSV操作

CSV（Comma Separated Values）是一种常用的文件格式，它用逗号来分隔不同的字段，通常用于存储和传输表格数据。在Python中，我们可以使用内置的csv模块来读取和写入CSV文件，进行各种操作。

1. CSV文件的读取

在Python中，要读取CSV文件，我们首先需要导入csv模块。然后，可以使用csv模块的reader()函数创建一个reader对象，该对象可以逐行读取CSV文件中的数据。


import csv
# 读取CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    
    # 遍历读取数据
    for row in reader:
        # 操作每一行的数据
        print(row)

上述代码中，我们使用了open()函数打开了一个名为data.csv的CSV文件，并创建了一个reader对象。然后，在一个for循环中遍历reader对象，每次都可以获得一行数据，存储在row变量中。

结果中的每一行数据都是一个列表，列表中的每个元素对应CSV文件中的一个字段。

2. CSV文件的写入

要写入CSV文件，我们同样需要导入csv模块，并使用csv模块的writer()函数创建一个writer对象。然后，可以使用writer对象的writerow()方法将数据写入CSV文件中。


import csv
# 写入CSV文件
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    
    # 写入数据
    writer.writerow(['Name', 'Age', 'Gender'])
    writer.writerow(['Alice', 25, 'Female'])
    writer.writerow(['Bob', 30, 'Male'])

上述代码中，我们首先使用open()函数创建一个名为data.csv的CSV文件，并创建了一个writer对象。然后，使用writerow()方法将数据按照列表的形式写入CSV文件中。

3. CSV文件的操作

在实际应用中，我们经常需要对CSV文件进行各种操作，例如筛选数据、计算均值等。下面是一些常见的操作示例。

3.1 数据筛选

假设我们有一个名为data.csv的CSV文件，其中包含了学生的成绩信息，包括姓名、年龄、数学成绩和语文成绩。我们可以使用Python来读取CSV文件，并筛选出数学成绩高于80分的学生。


import csv
# 读取CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    
    # 筛选数据
    for row in reader:
        name = row[0]
        age = int(row[1])
        math_score = int(row[2])
        chinese_score = int(row[3])
        
        if math_score > 80:
            print(f"{name}的数学成绩高于80分")

上述代码中，我们首先读取CSV文件，并在一个for循环中遍历文件中的每一行数据。然后，使用row变量获取每一行的数据，并将数值类型的数据转换成整型。最后，判断数学成绩是否高于80分，如果是，则打印相应的信息。

3.2 数据统计

除了筛选数据，我们还可以使用Python对CSV文件中的数据进行统计分析。下面是一个求平均值的示例。


import csv
# 读取CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    
    # 统计分析
    count = 0
    total_score = 0
    
    for row in reader:
        name = row[0]
        age = int(row[1])
        math_score = int(row[2])
        chinese_score = int(row[3])
        
        count += 1
        total_score += math_score + chinese_score
    
    average_score = total_score / (count * 2)
    
    print(f"平均分：{average_score}")

上述代码中，我们通过遍历CSV文件中的每一行数据，计算出总分和总人数，并根据公式计算出平均分数。最后，将平均分打印出来。

4. 总结

本文介绍了在Python中对CSV文件进行操作的方法，包括读取CSV文件、写入CSV文件以及对CSV文件进行筛选和统计分析。通过学习这些基本操作，我们可以更加灵活地处理CSV文件中的数据，实现各种需求。

通过使用csv模块，我们可以轻松地读取和写入CSV文件，处理其中的数据。并且，我们可以根据实际需求，对CSV文件进行筛选和统计分析，实现更复杂的操作。对于熟练掌握Python的开发者来说，CSV操作是一项强大且常用的技能，能够提高数据处理的效率，对于数据分析和科学计算等领域尤为重要。

python 下CSV 操作