📜  python 从 csv 创建字典 - Python (1)

📅  最后修改于: 2023-12-03 14:46:11.945000             🧑  作者: Mango

Python 从 csv 创建字典

在Python中,CSV(逗号分隔值)是一种常见的文件格式,用于存储表格数据。有时,我们需要将这些CSV文件转换为Python字典对象。在本文中,我们将学习如何在Python中通过CSV文件创建字典。

读取CSV文件

首先,我们需要使用Python的csv模块读取CSV文件。我们将使用Python的内置open()函数打开并读取CSV文件。以下是一个示例CSV文件:

name,age,location
John,25,Texas
Sarah,28,California
Mike,32,New York

现在,我们将使用csv模块中的DictReader()函数读取CSV文件并将其转换为字典。DictReader()函数将CSV文件的第一行作为字典的键,并将每行的值作为字典的值。

import csv

with open('example.csv', mode='r') as csv_file:
    csv_reader = csv.DictReader(csv_file)
    for row in csv_reader:
        print(row)

输出:

{'name': 'John', 'age': '25', 'location': 'Texas'}
{'name': 'Sarah', 'age': '28', 'location': 'California'}
{'name': 'Mike', 'age': '32', 'location': 'New York'}
创建字典

现在,我们已经将CSV文件读入了Python中的字典,我们可以将其用作我们需要的字典。我们可以使用Python的字典解析来选择我们需要的键和值。

以下是一个将CSV文件中的 name 作为键,age作为值的示例:

import csv

with open('example.csv', mode='r') as csv_file:
    csv_reader = csv.DictReader(csv_file)
    my_dict = {row['name']: row['age'] for row in csv_reader}

print(my_dict)

输出:

{'John': '25', 'Sarah': '28', 'Mike': '32'}

我们所做的就是使用Python的字典解析来选择CSV文件中的 name 列,并将它们作为我们新字典的键。同样地,我们选择 age 列,并将其作为我们新字典的值。

结论

在本文中,我们学习了如何在Python中通过CSV文件轻松创建字典。我们了解了如何使用Python的csv模块打开、读取和解析CSV文件,以及如何使用Python的字典解析来创建自己的字典。这使得我们能够轻松地将CSV文件转换为我们可以在Python中使用的字典对象。