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