在Python中比较对象时避免嵌套循环

Question

我得到了两段代码，这两段代码是由不同的人在不同的时间编写的，它们各自都能很好地工作。我的工作是比较这两段代码的输出。我还需要检查某些字段中是否有任何错误。你知道吗

脚本1返回一个Script1Object列表，每个都有自己的属性。类似地，脚本2返回一个Script2Object列表，这些列表与Script1Object相似，但不完全相同。你知道吗

我的代码是这样的：

script1_list = script1()
script2_list = script2()

for item1 in script1_list:
    for item2 in script2_list:
        if is_match(item1, item2):
            do_matching_action()
            break
        elif is_different_match(item1, item2):
            do_other_matching_action()
            break

    if is_bad(item1):
       do_error_action() 

其中is_match()匹配大约10个不同的属性。例如：

def is_match(item1, item2):
    return item1.name == item2.name and item1.ID == item2.number and item1.description.startswith(item2.desc)

和is_different_match()和is_bad()有一组类似的检查。你知道吗

这样做很好，但速度可能会很慢，尤其是当第一个列表可能有几十万个条目，而第二个列表可能有几千个条目时。代码也不是很模块化，所以我希望避免双for循环可以提高代码的可扩展性。我想知道如何改进这个实现，因为我确信这并没有充分利用Python的特性。你知道吗

Answer 1

假设匹配对象必须具有相似的id（Script1Object中的ID和Script2Object中的number），并且假设script2_list项都具有唯一的number，您可以使用dictionary创建从number到对象的映射。然后，当您进行迭代时，您可以通过字典的number直接从字典中获得一个Script2Object（而不必遍历整个script2_list）。然后可以像以前一样调用两个对象上的每个匹配函数：

script1_list = script1()
script2_list = script2()
script2_dict = {obj.number: obj for obj in script2_list}

for item1 in script1_list:
    item2 = script2_dict.get(item1.ID, None)
    if item2 is None: 
        print('no item2 found for this id: ', item1.ID)
        # do something if there's no matching id
    if is_match(item1, item2):
        do_matching_action()
    elif is_different_match(item1, item2):
        do_other_matching_action()

    if is_bad(item1):
       do_error_action()