python - 如何在匹配值(列表)包含 X 的字典中找到键？

Question

我有一个 python 字典，它将一些单词映射到其他单词列表。例如:

d = {"Hello": ["hi", "hello", "hey", "yo"],
     "Goodbye": ["bye", "see ya", "goodbye", "laters"]}

给定一个小写单词，我想检查该单词是否在任何字典值中并检索相应的键。

我确信有一些优雅的解决方案使用了一些 python 的功能(itertools 可能)，但它超出了我的能力范围......

有什么想法吗？

Answer 1

您可以匹配多个键，因此您需要生成一组(因为顺序无关紧要并且键是唯一的):

{key for key, words in d.iteritems() if search_word in words}

如果您只对第一个匹配感兴趣(因为您要保持单词的唯一性)，您可以使用 next() 和一个生成器表达式:

next((key for key, words in d.iteritems() if search_word in words), None)

如果您需要测试多个单词，您可能希望创建一个反向索引:

reverse_index = {}
for key, words in d.iteritems():
    for word in words:
        reverse_index.setdefault(word, set()).add(key)

之后你可以使用:

reverse_index.get(search_word, set())

得到相同的结果。

对于列表中的独特单词，反向索引将简单地为:

reverse_index = {word: key for key, words in d.iteritems() for word in words}
reverse_index.get(search_word)