python - 两组数据的匹配指标

Question

我有一个标题为 A B C D 的表格。D 下的值由 A、B、C 下的值索引。

我还有一个对象列表，这些对象由 A 和 B 列中包含的值索引，即 (A,B)。

对于每个对象，我想将表中与我的对象具有相同 A、B 索引的所有条目写入文件。

这就是我正在做的:

prescriptions = {}

#Open ABCD table and create a dictionary mapping A,B,C to D
with open(table_file) as table:
    reader = csv.reader(table, delimiter = '\t')
    for row in reader:
        code = (row[0], row[1], row[2])
        prescriptions[code]=row[3]

for x in objects:
    x_code = (x.A, x.B)

    for p in prescriptions:
        #check to see if A,B indices on x match those of the table entry
        if p[0:2] == x_code:
            row = prescriptions[p]
            line = ",".join(p) +"," + row +"\n"
            output.write(line)

这行得通。我得到了我想要的确切输出；然而，当表格和列表变大时，会花费大量时间。

我想修改我的迭代器(当我找到一个匹配项时删除 p)，but I know not to do that .

我能做些什么来加快速度吗？

Answer 1

我猜 prescription 是一本字典？

为什么不用字典 prescription2 以 A、B 作为键、C、D 的列表作为值？这将使您免于遍历所有字典的麻烦。

prescriptions = {}
prescriptions2 = {}

#Open ABCD table and create a dictionary mapping A,B,C to D
with open(table_file) as table:
    reader = csv.reader(table, delimiter = '\t')
    for row in reader:
        code = (row[0], row[1], row[2])
        prescriptions[code]=row[3]
        key = (row[0],row[1])
        if not key in prescription2:
            prescription2[key] = []
        value = (row[2],row[3])
        prescription2[key].append(value)

for x in objects:
    x_code = (x.A, x.B)
    if x_code in prescription2:
        for item in prescription2[x_code]:
            line = ",".join(x_code+item)+"\n"
            output.write(line)