在 Python 中将 CSV 转换为字典

CSV（Comma-Separated Values）是一种常见的数据格式，它由逗号分隔的值构成。在 Python 中，我们可以将 CSV 文件转换为字典，方便数据处理和分析。

读取 CSV 文件

我们可以使用 Python 内置的 csv 模块读取 CSV 文件。假设我们有一个叫做 data.csv 的文件，内容如下：

name,age,gender
Alice,25,Female
Bob,30,Male
Charlie,35,Male

我们可以使用以下代码读取该文件：

import csv

with open('data.csv', newline='') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        print(row)

这段代码首先用 open 函数打开文件，并将文件对象传给 csv.DictReader 函数。DictReader 函数会根据文件内容生成一个字典迭代器。我们可以遍历该迭代器并输出每一行的字典内容。

输出结果

运行上面的代码，输出结果如下：

{'name': 'Alice', 'age': '25', 'gender': 'Female'}
{'name': 'Bob', 'age': '30', 'gender': 'Male'}
{'name': 'Charlie', 'age': '35', 'gender': 'Male'}

每一行都变成了一个字典，字典的键是 CSV 文件的第一行，即列名。

如果我们想进一步处理这些数据，比如只输出男性的信息，我们可以在 for 循环中添加条件判断：

import csv

with open('data.csv', newline='') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        if row['gender'] == 'Male':
            print(row)

这段代码只输出 gender 是 Male 的行：

{'name': 'Bob', 'age': '30', 'gender': 'Male'}
{'name': 'Charlie', 'age': '35', 'gender': 'Male'}

总结

在 Python 中，我们可以使用 csv 模块轻松读取和处理 CSV 文件。将 CSV 文件转换为字典的方法是使用 csv.DictReader 函数。该函数会返回一个字典迭代器，我们可以遍历并进一步处理其中的每一行数据。