Как достать данные из такого словаря?

Question

[Артём Иннокентьев]

Есть словарь:
data ={'a': {'b': {'c': 1}}}
Надо достать значение ключа 'c' с помощью такой конструкции data[key], при этом такой вариант не подходит data['a']['b']['c']. Как это сделать?

Comments

[Roman K]

Как это не подходит?

[Артём Иннокентьев]

Это сработает, но это не то, что нужно.

[sim3x]

Артём Иннокентьев: почему?

[Артём Иннокентьев]

Надо достать значение ключа 'c' с помощью такой конструкции data[key]. Отлично подходит реализация angru

Answer 1

[angru]

Python 3. но это костыль конечно же.

from functools import reduce


class CustomDict(dict):
    def __getitem__(self, key):
        return reduce(lambda acc, x: acc.__getitem__(x), key.split('.'), super())


data = CustomDict({'a': {'b': {'c': 1}}})

print(data['a.b.c'])

Comments

[angru]

естественно действует только со строковыми ключами

[Roman K]

Это круто, а не костыль.

[angru]

Roman Kitaev: ну, боевой код так не пишут, я подумал автору надо чисто в академических целях, вот и привел пример, но таких вещей стоит остерегаться.

[Артём Иннокентьев]

А что надо сделать для боевого кода?

[Roman K]

angru: Я про саму реализацию.

[angru]

Артём Иннокентьев: чтобы все более менее работало надо еще переопределить методы CustomDict.__setitem__, CustomDict.get, добавить проверку строковый ключ или нет, если нет, то возвращать просто super().__getitem(key), но это подход чреват различными трудноотлавливаемыми ошибками, так что лучше вообще не трогать словарь, а написать внешний метод:

def get_multi_key(d, key):
    if isinstance(key, str):
        return reduce(lambda acc, x: acc[x], key.split('.'), d)
    else:
        return d[key]


print(get_multi_key(data, 'a.b.c'))
print(get_multi_key(data, 1))

Answer 2

[Антон Федорян]

Смысл тот же, что и

data['a']['b']['c']

((data.get('a')).get('b')).get('c')

Answer 3

[lyeskin]

https://github.com/akesterson/dpath-python