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

时间:2015-09-26 01:58:53

标签: python list python-3.x dictionary

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]

1 个答案:

答案 0 :(得分:4)

您还需要迭代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)操作,而在列表中搜索是上) )。示例 -

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