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

时间:2021-06-07 10:05:21

标签: python python-3.x dictionary dictionary-comprehension

我有一本字典,键和其他键几乎相似。我需要使用几乎相似的键提取所有值。 示例:

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

这里我们有一个 cd1ef1, cd2ef2, cd3ef3 的映射。

我需要的结果是所有 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 的所有值 但第二种方法的问题是我无法将 cd1ef1 映射。

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

2 个答案:

答案 0 :(得分:0)

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

答案 1 :(得分:0)

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

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

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