2

我有一个标题为 ABC 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)

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

我很想修改我的迭代器（找到匹配时删除 ap），但我知道不要这样做。

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

4

1 回答 1

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)
于 2013-06-24T22:35:24.827 回答