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

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

我需要检查"list"上的值是否与字典内列表中的值匹配.这甚至可能吗？

我尝试使用它,但我只得到一个空列表:

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