有没有办法将ID相似的文件添加到字典中

Question

我正在从一个文件中检索成绩，我想将它们插入到一个以id为关键字的字典中，最好的方法是什么？下面是代码。

from HW08_Swayam_Shah import file_reader
import os
from collections import defaultdict
def grades(path):
l= defaultdict()
g = {"A":4.0,"A-":3.75,"B+":3.25,"B":3.0,"B-":2.75,"C+":2.25,"C-":0,"D+":0,"D":0,"D-":0,"F":0}
for id, course, grade, prof_id in file_reader(
os.path.join(path, "g.txt"), fields=4, sep='|', header=False):
for k,v in g.items():
if grade==k:
l[id].append(v)
return l
x = grades("C://Users/Documents/Downloads")
print(x)

以下是我正在使用的输入文件：

10103|SSW 567|A|98765
10103|SSW 564|A-|98764
10103|SSW 687|B|98764

正如您所看到的，第一个字段对于所有列都是相似的，但我需要它作为我的键。显然，字典会抛出一个键错误，但当键相同时，我希望它将其添加到下一个值中。类似于：

{10103:{A,A-,B}}

我怎样才能做到这一点？

Answer 1

您需要指定defaultdict:的工厂

l = defaultdict(list)

如果字典项不存在，这将创建一个空列表，那么append()将起作用。

您不需要在字典g上循环。由于grade是一个键，只需使用g[grade]即可获得值。

def grades(path):
l= defaultdict(list)
g = {"A":4.0,"A-":3.75,"B+":3.25,"B":3.0,"B-":2.75,"C+":2.25,"C-":0,"D+":0,"D":0,"D-":0,"F":0}
for id, course, grade, prof_id in file_reader(
os.path.join(path, "g.txt"), fields=4, sep='|', header=False):
grade_numeric = g[grade]
l[id].append(grade_numeric)
return l