使用DictReader访问CSV的第三列时获取KeyError

Question

我尝试使用Python按日期对csv文件进行排序(最新条目优先(。csv文件中包含日期的列是csv文件的第三列。

当我运行下面的代码时，我得到以下错误消息：

KeyError:2

你能给我提供指导方针来更正下面的脚本吗：

import csv
import os
from csv import writer
from datetime import datetime
with open("data.csv", newline="") as csvfile:
spamreader = csv.DictReader(csvfile, delimiter=",")
sortedlist = sorted(spamreader, key=lambda row: datetime.strptime(row[2]), reverse=False)
with open("sortdata.csv", "w") as f:
fieldnames = ["detection_date"]
writer = csv.DictWriter(f, fieldnames=fieldnames)
writer.writeheader()
for row in sortedlist:
writer.writerow(row)

Answer 1

刚读到错误，keyerror: 2，我的眼睛就转到了datetime.strptime(row[2])。

您正试图为DictReader使用基于0的列索引号，它需要一个列名(它派生自文件中的第一行数据，它希望它是一个"；标题"(。

所以，把row[2]换成row['third_column_name']，你应该很好。

如果没有标头，请使用csv.reader()。