python - 创建一个新的字典值列表，其键与单独列表中的项目匹配

Question

该字典将农场动物存储为键，将它们的位置存储为值。

id_name_dict = {
    'cat': 'barn', 'dog': 'field', 'chicken': 'coop',
    'sheep': 'pasture', 'horse': 'barn', 'cow': 'barn'
}

此列表存储我想知道其位置的农场动物的名称

wanted_farm_animals = ['cat,', 'dog', 'horse']

所需的输出是一个包含wanted_farm_animals位置的新列表

n = ['barn', 'field', 'barn']

这是我尝试执行此操作的代码

n = []
for animal, location in id_name_dict.items():
    for a in wanted_farm_animals:
        if a == animal:
            n.append(location)
print(n)

但是，输出并不完整。只是

['field', 'barn']

如何获得正确的所需输出？

Answer 1

简单的列表理解怎么样？

id_name_dict = {
    'cat': 'barn', 'dog': 'field', 'chicken': 'coop',
    'sheep': 'pasture', 'horse': 'barn', 'cow': 'barn'
}    
wanted_farm_animals = ['cat', 'dog', 'horse']

result = [v for k,v in id_name_dict.items() if k in wanted_farm_animals]
# ['barn', 'field', 'barn']