python - 根据其他两个词典中的匹配术语创建词典 - Python

Question

我正在尝试比较描述产品目录内容的两个大型词典。每个字典都包含一个唯一的编码键和每个键的术语列表。

dict1 = {
"SKU001": ["Plumbing", "Pumps"], 
"SKU002": ["Motors"], 
"SKU003": ["Snow", "Blowers"], 
"SKU004": ["Pnuematic", "Hose", "Pumps"],
...
}

dict2 = {
"FAS001": ["Pnuematic", "Pumps"], 
"GRA001": ["Lawn", "Mowers"], 
"FAS002": ["Servo", "Motors"], 
"FAS003": ["Hose"], 
"GRA002": ["Snow", "Shovels"], 
"GRA003": ["Water", "Pumps"]
...
}

我想创建一个新字典，它从 dict1 中借用键，其值是 dict2 中的键列表，其中至少一个词值匹配。理想的最终结果可能类似于:

match_dict = {
"SKU001": ["FAS001", "GRA003"], 
"SKU002": ["FAS002"], 
"SKU003": ["GRA002"], 
"SKU004": ["FAS001", "FAS003", "GRA003], 
...
}

虽然我在创建此输出时遇到了问题。是否可以创建键列表并将其作为值分配给另一个键？我已经尝试使用如下所示的嵌套循环进行了几次尝试，但输出不符合预期，我不确定它是否能正常工作。感谢您的帮助!

matches = {}
for key, values in dict1.items():
    for value in values:
        if value in dict2.values():
            matches[key] = value
print(matches)

Answer 1

这是一种可能的实现方式:

dict1 = {
    "SKU001": ["Plumbing", "Pumps"], 
    "SKU002": ["Motors"], 
    "SKU003": ["Snow", "Blowers"], 
    "SKU004": ["Pnuematic", "Hose", "Pumps"],
}
dict2 = {
    "FAS001": ["Pnuematic", "Pumps"], 
    "GRA001": ["Lawn", "Mowers"], 
    "FAS002": ["Servo", "Motors"], 
    "FAS003": ["Hose"], 
    "GRA002": ["Snow", "Shovels"], 
    "GRA003": ["Water", "Pumps"]
}
match_dict_test = {
    "SKU001": ["FAS001", "GRA003"], 
    "SKU002": ["FAS002"], 
    "SKU003": ["GRA002"], 
    "SKU004": ["FAS001", "FAS003", "GRA003"], 
}

# Find keys for each item in dict2
dict2_reverse = {}
for k, v in dict2.items():
    for item in v:
        dict2_reverse.setdefault(item, []).append(k)
# Build dict of matches
match_dict = {}
for k, v in dict1.items():
    # Keys in dict2 associated to each item
    keys2 = (dict2_reverse.get(item, []) for item in v)
    # Save sorted list of keys from dict2 without repetitions
    match_dict[k] = sorted(set(k2i for k2 in keys2 for k2i in k2))
# Check result
print(match_dict == match_dict_test)
# True