在字典中查找相似关键字列表

Question

我有一本字典，里面有几乎相似的关键字和其他关键字。我需要用几乎相似的键提取所有的值。示例：

example_dictionary = {'ab1':'1', 'ab2':'2','ab3':'3', 'cd1':'4', 'cd2':'5', 'cd3':'6', 'ef1':'7', 'sdsf':'8', 'ad':'9', 'ef2':'10', 'ef3':'11'}  
keymatch = ['ab','cd','ef']

这里我们有一个cd1与ef1, cd2、ef2, cd3与ef3的映射。

我需要的结果是所有ab1, ab2, ab3, cd1, cd2, cd3, ef1, ef2, ef3.的值

我试过了：

abList = [val for key, val in example_dictionary.items() if keymatch[0] in key]

这为我提供了ab1, ab2, ab3的所有值

我试过了：

result = [val for key, val in example_dictionary.items() for item in keymatch if item in key]

这给了我ab1, ab2, ab3, cd1, cd2, cd3, ef1, ef2, ef3的所有值，但是第二种方法的问题是我不能将cd1映射到ef1。

有没有人能提供一个更好的方法？

Answer 1

IIUC你想要有一个合并了“相似关键字”的字典。您可以通过将字典理解与列表理解相结合来做到这一点。

{km: [v for k, v in example_dictionary.items() if km in k] for km in keymatch}

{'ab': ['1', '2', '3'], 'cd': ['4', '5', '6'], 'ef': ['7', '10', '11']}

Answer 2

您可以使用名为startswith()的字符串函数。代码如下：

result = []
for k, v in example_dictionary.items(): 
    for item in keymatch: 
        if k.startswith(item): 
            result.append(v)

您可能会发现这个tutorial for startswith很有用。