按标题标题解析 CSV 文件

Question

我有几个 CSV 文件，其中包含有关不同用户的信息。 每个文件都有 3 或 4 列，第一行是标题和其下的相应信息。 如：

Username  FirstName  LastName  Occupation
JSmith12  John       Smith     Staff

有些列在 CSV 文件的一半中乱序或丢失，因此我不能简单地执行row.split(',')[0]来获取所有 CSV 文件中的所有用户名。

我的问题是：我如何能够通过标题而不是行来获取和读取所有列信息。

Answer 1

查看使用DictReader类：

$ cat test.csv
Username,FirstName,LastName,Occupation
JSmith12,John,Smith,Staff
$ python
>>> import csv
>>> 
>>> with open("test.csv") as f:
...     reader = csv.DictReader(f)
...     for row in reader:
...         print(row)
... 
{'Username': 'JSmith12', 'LastName': 'Smith', 'FirstName': 'John', 'Occupation': 'Staff'}

每一row都是一个字典，其中的键是从 csv 文件头中提取的。

Answer 2

Python 标准库中的csv库包含一个名为DictReader的类，它可以执行您想要的操作。