Python中csv模块的基本使用教程

下面是关于Python中csv模块基本使用教程的完整攻略。

什么是csv模块

CSV即“Comma-separated values”，指用逗号隔开的文本数据，是一种行业通用的轻量级数据交换格式。csv模块是Python标准库中的一个模块，它提供了读写csv文件的功能。使用csv模块可以方便地处理csv格式数据，例如读取、写入、解析数据等。

csv模块的基本使用

以下是csv模块的一些基本用法。

读取csv文件数据

使用csv模块可以很方便地读取csv文件中的数据。假设我们有一个名为“data.csv”的csv文件，它包含以下数据：

Name, Age, Gender
Alice, 18, Female
Bob, 22, Male
Charlie, 25, Male

接下来用csv模块读取它：

import csv

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

上述代码会逐行读取data.csv文件中的数据，并输出到控制台上。输出结果为：

['Name', ' Age', ' Gender']
['Alice', ' 18', ' Female']
['Bob', ' 22', ' Male']
['Charlie', ' 25', ' Male']

注意，在读取csv文件时，csv模块会默认将每行数据转换成列表形式，以便后续处理。

写入csv文件数据

使用csv模块可以很方便地写入数据到csv文件中。下面是一个使用csv模块写入数据的示例：

import csv

with open('data.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age', 'Gender'])
    writer.writerow(['Alice', '18', 'Female'])
    writer.writerow(['Bob', '22', 'Male'])
    writer.writerow(['Charlie', '25', 'Male'])

上述代码会创建一个名为“data.csv”的文件，并将数据写入其中。csv.writer()函数会将每个元素以逗号分隔的形式写入文件中，最后生成csv格式的数据，如下所示：

Name,Age,Gender
Alice,18,Female
Bob,22,Male
Charlie,25,Male

其他用法

在csv模块中，还有一些其他的用法，例如：

• csv.DictReader()函数：用于读取csv文件中的数据并返回一个字典列表。
• csv.DictWriter()函数：用于将字典列表写入到csv文件中。

关于csv模块的更多用法，可以参考Python官方文档。

示例说明

接下来通过两个示例来说明csv模块的使用。

示例一：统计数据

假设我们有一个名为“score.csv”的csv文件，它包含以下数据：

Name,Chinese,Math,English
Alice,80,90,85
Bob,75,88,92
Charlie,82,77,81

接下来我们使用csv模块读取这个文件，并对每个学生的成绩进行统计：

import csv

scores = {}

with open('score.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        name = row['Name']
        chinese = int(row['Chinese'])
        math = int(row['Math'])
        english = int(row['English'])

        total = chinese + math + english
        average = round(total / 3, 2)

        scores[name] = {'Chinese': chinese, 'Math': math, 'English': english, 'Total': total, 'Average': average}

print(scores)

上述代码会将每个学生的成绩统计出来，并按照学生姓名将其存储到字典中。输出结果为：

{'Alice': {'Chinese': 80, 'Math': 90, 'English': 85, 'Total': 255, 'Average': 85.0}, 
 'Bob': {'Chinese': 75, 'Math': 88, 'English': 92, 'Total': 255, 'Average': 85.0}, 
 'Charlie': {'Chinese': 82, 'Math': 77, 'English': 81, 'Total': 240, 'Average': 80.0}}

示例二：保存数据

假设我们有一些数据需要保存到csv文件中。接下来我们使用csv模块将这些数据写入到“data.csv”文件中：

import csv

data = [['Name', 'Age', 'Gender'], ['Alice', '18', 'Female'], ['Bob', '22', 'Male'], ['Charlie', '25', 'Male']]

with open('data.csv', 'w') as file:
    writer = csv.writer(file)
    for row in data:
        writer.writerow(row)

上述代码会将data列表中的数据写入到csv文件中，并生成如下文件：

Name,Age,Gender
Alice,18,Female
Bob,22,Male
Charlie,25,Male

通过上面两个示例，我们可以看到，使用csv模块可以很方便地对csv数据进行处理。因此，使用csv模块是进行数据处理的常用方法之一。