python - 在字典中查找相似键的列表

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']

这里我们有一个cd1withef1, cd2和ef2, cd3with的映射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.

有人可以提供更好的方法吗？

标签： pythonpython-3.xdictionarydictionary-comprehension

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']}