8

我有一个这样的列表:

lis = ['Date', 'Product', 'Price']

我想将其与以下内容进行比较:

dict = {'Date' : '2013-05-01', 'Salary' : '$5000', 'Product' : 'Toys', 'Price' : '$10', 'Salesman' : 'Smith'}

我想将列表的每个项目与字典的键进行比较并制作一个新字典。
我尝试过的是:

n = {}
for k,v in dict.items():
    for i in lis:
        if i==k:
            n[k] = v

输出:

n = {'Date' : '2013-05-01', 'Product' : 'Toys', 'Price' : '$10'}

这行得通,但我想通过生成器来做——有人可以帮我做吗?

4

3 回答 3

13

而是将其视为lis一个集合,因此您可以使用字典视图和交集:

# python 2.7:
n = {k: d[k] for k in d.viewkeys() & set(lis)}

# python 3:
n = {k: d[k] for k in d.keys() & set(lis)}

或者,您可以使用简单的 dict 理解和in测试d

# python 2.6 or older:
n = dict((k, d[k]) for k in lis if k in d)

# python 2.7 and up:
n = {k: d[k] for k in lis if k in d}

这假定并非所有值 inlis都会在d; 如果它们总是存在,则if k in d可以放弃测试。

对于您的具体情况,第二种形式要快得多:

>>> from timeit import timeit
>>> timeit("{k: d[k] for k in d.viewkeys() & s}", 'from __main__ import d, lis; s=set(lis)')
2.156520128250122
>>> timeit("{k: d[k] for k in lis if k in d}", 'from __main__ import d, lis')
0.9401540756225586
于 2013-03-13T12:28:06.337 回答
1
filtered_dict = dict((k, original_dict[k]) for k in lis if k in original_dict)

或者,如果您有 2.7+:

filtered_dict = {k: original_dict[k] for k in lis if k in original_dict}

如果要使用生成器:

item_generator = ((k, original_dict[k]) for k in lis if k in original_dict)

生成器将产生(key, value)对。

于 2013-03-13T12:28:56.667 回答
0

对于 Python 3.6

dict_keys_list = list(dict.keys())
dict_keys_list.sort()
for key in dict_keys_list:
     if key in lis:
          print("Key is present")

您希望将键列表与字典键进行比较的另一种方法。

person = { 'name' : 'Don', 'nickname' : 'John', 'age' : 45, 'salary' : 50000, 'place' : 'Monaco, Europe' }
compare_list = ['nickname', 'place', 'name', 'salary', 'age']
person_keys_list = list(person.keys())
person_keys_list.sort()
compare_list.sort()
person_keys_list == compare_list # Gives you True
于 2021-04-19T14:04:35.487 回答