python - 如何根据python中的 header 提取某些csv数据

Question

我如何根据 python 中的 header 从 csv 文件中提取特定数据？ 例如，假设 csv 文件包含以下信息:

Height,Weight,Age
6.0,78,25

我怎样才能在 python 中只检索年龄？

Answer 1

我支持 csv 推荐，但我认为这里使用 csv.DictReader 会更简单:

(Python 2):

>>> import csv
>>> with open("hwa.csv", "rb") as fp:
...     reader = csv.DictReader(fp)
...     data = next(reader)
...     
>>> data
{'Age': '25', 'Weight': '78', 'Height': '6.0'}
>>> data["Age"]
'25'
>>> float(data["Age"])
25.0

在这里，我使用 next 只是为了获取第一行，但如果您愿意，您可以遍历这些行和/或提取完整的信息列。