Как обрабатывать многоуровневые словари в Python?

Question

[Андрей]

Можно как-то более оптимально записать/обработать такую конструкцию при обходе dict:

try:
    new_dic['Name_1'] = source_1['key_1']['key_2']['key_3']
except:
    new_dic['Name_1']  = None
try:
    new_dic['Name_2'] = source_2['key_1']['key_2']
except:
    new_dic['Name_2']  = None
try:
    new_dic['Name_3'] = source_1['key_1']['key_2']['key_3']['key_4']
except:
    new_dic['Name_3']  = None
...

Answer 1

[Alexeytur]

def dict_get_recursive(source, path):
	current_dict = source
	for path_idx, path_key in enumerate(path):	
		if path_idx == len(path)-1:
			return current_dict.get(path_key, None)
		
		current_dict = current_dict.get(path_key, None)
		if current_dict is None:
			return None
			
new_dic['Name_1'] = dict_get_recursive(source_1,['key_1','key_2','key_3'])
new_dic['Name_2'] = dict_get_recursive(source_2,['key_1','key_2'])
new_dic['Name_3'] = dict_get_recursive(source_1,['key_1','key_2','key_3','key_4'])

Comments

[Андрей]

Alexeytur а если один из key - это list? (например source_1['key_1'][0]['key_3']['key_4'])

[Alexeytur]

Андрей,
list или int как у вас source_1['key_1'][0]['key_3']['key_4']?
В питоне ключами словарей могут быть только неизменяемые объекты, list не может быть ключом.

[Андрей]

Alexeytur, это понятно, не правильно выразился, если ключ key_1 а value у этого ключа список словарей [{'key_1_1': 'val'}, {'key_1_2': 'val'}, {'key_1_3': 'val'}], в данном случае вышеуприведенная рекурсия не работает...

[Alexeytur]

Андрей,
Можно доработать функцию, чтобы current_dict мог быть списком, а не только словарем