Python вложенный словарь поиска со значениями по умолчанию

Question

>>> d2
{'egg': 3, 'ham': {'grill': 4, 'fry': 6, 'bake': 5}, 'spam': 2}
>>> d2.get('spamx',99)
99
>>> d2.get('ham')['fry']
6

Я хочу получить значение fry внутри ветчины, если нет, получить значение, 99 или 88, как 2-й пример.Но как?

Answer 1

d2.get('ham', {}).get('fry', 88)

Я бы, вероятно, разбил его на несколько утверждений в реальной жизни.

ham = d2.get('ham', {})
fry = ham.get('fry', 88)

Answer 2

Если вам нужно сделать это много, вы можете написать вспомогательную функцию

def get_nested(d, list_of_keys, default):
    for k in list_of_keys:
        if k not in d: 
            return default
        d=d[k]
    return d

print get_nested(d2,['ham','spam'],99)
print get_nested(d2,['ham','grill'],99)

Answer 3

Чтобы значения по умолчанию get работали правильно, первым значением по умолчанию должен быть словарь, чтобы вы могли правильно связать вызовы .get в случае первого сбоя.

d.get('ham',{}).get('fry',88)

Вы также можете использовать попытку, кроме блока

def get_ham_fry()
  try:
    return d['ham']['fry']
  except AttributeError,e:
    return 88

Answer 4

Вот решение для работы с вложенными словарями:

def get(root, *keys):
    """
    Returns root[k_1][k_2]...[k_n] if all k_1, ..., k_n are valid keys/indices. 
    Returns None otherwise
    """
    if not keys:
        return root
    if keys[0] not in root:
        return None
    if keys[0] in root:
        return get(root[keys[0]], *keys[1:])

Использование:

>>> d = {'a': 1, 'b': {'c': 3}}
>>> get(d, 'b', 'c')
3
>>> get(d. 'key that's not in d')
None
>>> get(d)
{'a': 1, 'b': {'c': 3}}