Сравнение списка со словарем для создания нового списка в python

Question

У меня есть один список и один словарь. Я хочу сравнить значения списка с ключами словаря. Если у меня есть:

mydict = {'Hello':1,'Hi':2,'Hey':3} 

и:

mylist = ['Hey','What\'s up','Hello'] 

Я хочу выход быть:

output = [3, None, 1] 

Спасибо!

Я пробовал [mydict[i] for i in mylist], и вместо этого у меня есть ошибка. Затем я попытался использовать вложенные для циклов (я удалил этот бит), но я решил, что это было неэффективно.

Answer 1

Используйте list comprehension:

output = [ mydict.get(key) for key in mylist ] 

Обратите внимание, что dict.get возвращает None если ключ не в Словаре.

Answer 2

Использование dict.get(), который по умолчанию None если ключ не существует:

[mydict.get(k) for k in mylist] 

---

>>> mydict = {'Hello':1,'Hi':2,'Hey':3} 
>>> mylist = ['Hey','What\'s up','Hello'] 
>>> out = [mydict.get(k) for k in mylist] 
>>> out 
[3, None, 1]