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

Question

У меня есть вложенный dict, как указано ниже:

d = {'A': [{'B': [{'C': [{'D1':[]}, {'D2': []}]}]}]} # just an example

И мне дан список / путь, подобный этому:

l = ['A','B','C','D1']

Я бы хотел пройти попуть в dict для получения соответствующего значения D1.Для этого я написал следующую программу:

def find_dict(ld, key):
    # we can assume that 'ld' has dicts each of which has unique key
    for d in ld:
        if key not in d:
            continue
        return d

def walk_dict(d,path):
    temp = None
    for i,n in enumerate(path):
        if i == 0:
            temp = d.get(n)
        elif i < (len(path)-1):
            temp = find_dict(temp, n)
            temp = temp.get(n)
        else: # last item
            temp = find_dict(temp, n)
            print('Found it!')
            print(i,n)
            print(temp.get(n))
            return temp.get(n)

d = {'A': [{'B': [{'C': [{'D1':[]}, {'D2': []}]}]}]}
l = ['A','B','C','D1'] # successfully retrieved the value of 'D1'
walk_dict(d,l)
l = ['A','B','C','D2'] # successfully retrieved the value of 'D2'
walk_dict(d,l)
l = ['A','B','C','D3'] # get NoneType error as expected because 'D3' does not exist
walk_dict(d,l)

Я хотел бы знать, есть ли лучший способ сделать это.Я чувствую, что этот подход слишком многословен, и, возможно, даже имеются скрытые ошибки.

Заранее благодарю за ваши ответы / предложения!

Answer 1

Я использовал рекурсивный подход.

d = {'A': [{'B': [{'C': [{'D1':'xyz'}, {'D2': []}]}]}]}
l = ['A','B','C','D1']
output = []
def find_path(dict_obj,key=0):
    for k,v in dict_obj.items():
        if k == l[key]:
            if key == len(l)-1:
                output.append(v)
                return
            if isinstance(v,dict):
                find_path(v, key+1)
            elif isinstance(v,list):
                for i,item in enumerate(v):
                    if isinstance(item,dict):
                        find_path(item,key+1)

find_path(d)
print(output[0] if output else None)

Выходы:

l = ['A','B','C','D1']
# xyz
l = ['A','B','C']
# [{'D1': 'xyz'}, {'D2': []}]
l = ['A','B','C','D3']
# None
l = ['A','B','C','D2']
# []

Answer 2

Вы можете использовать эту рекурсивную функцию:

d = {'A': [{'B': [{'C': [{'D1':[]}, {'D2': []}]}]}]} # just an example
def walk(_d, path):
   a, *b = path
   if isinstance(_d, list):
     _d = [i for i in _d if a in i]
     if not _d:
       return None
     _d = _d[0]
   return _d[a] if not b else walk(_d[a], b)

tests = [[['A','B','C','D1'], []], [['A','B','C','D2'], []], [['A','B','C','D3'], None]]
for a, b in tests:
  assert walk(d, a) == b

print('all cases passed')

Вывод:

all cases passed

Ниже приведены временные параметры для текущих ответов.Исходный код для синхронизации и настройки ввода можно найти в виде здесь .

Answer 3

Как насчет того, чтобы продиктовать это следующим образом:

from copy import deepcopy

def walk_dict(ld, path):
    if isinstance(path, str):
        path = path.split()
    t = deepcopy(ld)
    for key in path:
        if isinstance(t, list):
            try:
                t = [d for d in t if key in d][0]
            except IndexError:
                return
        try:
            t = t[key]
        except KeyError:
            return
    else:
        return t

def main():
    d = {'A': [{'B': [{'C': [{'D1':['values of D1']}, {'D2': "I'm D2"}]}]}]}
    a = 'A B C D1'
    print(walk_dict(d, a))
    a = 'A B C D2'
    print(walk_dict(d, a))
    a = 'A B C D3'
    print(walk_dict(d, a))
    a = 'A B1 C D3'
    print(walk_dict(d, a))


if __name__ == '__main__':
    main()

Вывод:

['values of D1']
I'm D2
None
None