搜索列表中的值是否在格式为 key-string, value-list(strings) 的字典中

Question

my_dict = {                              # This dictionary is generated thru
'a' : [ 'value1', 'value4', 'value5' ],  # the info given by the user
'b' : [ 'value2', 'value6', 'value7'],
'c' : [ 'value3', 'value8', 'value9']
}

list = [ 'value1', 'value2' ] # List is generated using list comprehension

我需要生成一个将输出如下内容的列表：

output_list = ['a', 'b']

我需要检查“列表”中的值是否与字典内列表中的值匹配。这甚至可能吗？

我尝试使用它，但我只得到一个空列表：

[key for key, value in my_dict.items() if value in list]

Answer 1

您还需要遍历list（并且您不应该使用list 作为变量名，它会影响内置的list 函数）。示例 -

[key for item in lst for key,value in my_dict.items() if item in value]

演示 -

>>> my_dict = {                              # This dictionary is generated thru
... 'a' : [ 'value1', 'value4', 'value5' ],  # the info given by the user
... 'b' : [ 'value2', 'value6', 'value7'],
... 'c' : [ 'value3', 'value8', 'value9']
... }
>>>
>>> lst = [ 'value1', 'value2' ]
>>> [key for item in lst for key,value in my_dict.items() if item in value]
['a', 'b']

---

如果您使用set 而不是list 将值存储在字典中，您可以获得更好的性能（因为在集合内搜索是 O(1) 操作，而在列表内搜索是 O(n) ）。示例 -

my_dict = {key:set(value) for key,value in my_dict.items()}
[key for item in lst for key,value in my_dict.items() if item in value]

演示 -

>>> my_dict = {key:set(value) for key,value in my_dict.items()}
>>> pprint(my_dict)
{'a': {'value4', 'value5', 'value1'},
 'b': {'value6', 'value7', 'value2'},
 'c': {'value3', 'value9', 'value8'}}
>>> lst = [ 'value1', 'value2' ]
>>> [key for item in lst for key,value in my_dict.items() if item in value]
['a', 'b']

---

如果您尝试检查列表中的任何值是否与字典中列表中的任何值匹配，您可以使用set.intersection 并检查结果是否为空。示例 -

[key for key, value in my_dict.items() if set(value).intersection(lst)]

这个结果不会被排序，因为字典没有任何特定的顺序。

演示 -

>>> my_dict = {
... 'a' : [ 'value1', 'value4', 'value5' ],
... 'b' : [ 'value2', 'value6', 'value7'],
... 'c' : [ 'value3', 'value8', 'value9']
... }
>>> lst = [ 'value1', 'value2' ]
>>> [key for key, value in my_dict.items() if set(value).intersection(lst)]
['b', 'a']