使用Python将 CSV 数据加载到列表和字典中

CSV（Comma-Separated Values）是一种常见的数据格式，以逗号分隔各个字段，一行数据表示一个记录。在Python中，我们可以使用内置的csv模块来读取CSV文件，并将数据分别保存到列表和字典中。

加载CSV文件到列表中

首先，我们需要使用csv模块打开CSV文件。以下是一个名为data.csv的示例文件：

Name, Age, Gender
Alice, 25, Female
Bob, 30, Male
Charlie, 35, Male

以下是使用csv模块将数据加载到列表的示例代码：

import csv

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

print(data)

输出：

[
    ['Name', ' Age', ' Gender'],
    ['Alice', ' 25', ' Female'],
    ['Bob', ' 30', ' Male'],
    ['Charlie', ' 35', ' Male']
]

每一行数据都作为列表中的一个元素。在with语句块中打开CSV文件后，我们使用csv.reader()函数创建一个读取器对象。读取器对象可以迭代地获取每一行数据，并使用列表推导式将它们保存到data变量中。

加载CSV文件到字典中

我们也可以将CSV文件的每一行数据保存为一个字典，以字段名作为键，字段值作为值。以下是一个示例代码：

import csv

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

print(data)

输出：

[
    {'Name': 'Alice', ' Age': '25', ' Gender': 'Female'},
    {'Name': 'Bob', ' Age': '30', ' Gender': 'Male'},
    {'Name': 'Charlie', ' Age': '35', ' Gender': 'Male'}
]

在这个例子中，我们使用csv.DictReader()函数创建一个读取器对象，并传入参数fieldnames，指定了每个字段的名称。读取器对象可以迭代地获取每一行数据，并使用列表推导式将它们保存到data变量中。

结论

使用Python将CSV数据加载到列表和字典中非常容易。在本文中，我们介绍了如何使用csv模块读取CSV文件，并将数据保存到列表和字典中。对于大量的数据处理任务，Python提供了强大的库和工具，使数据处理任务变得更加轻松。