如何潛入堆疊字典優雅

Question

比方說，我有一個像

{ 
    "key_a": "value", 
    "key_b": { 
     "key_b_a": "value", 
     "key_b_b": { 
      "key_b_b_a": "value" 
     } 
    } 
} 

的字典我想是創建到刪除給定鍵或更改其值的方法。

def del_key(key): 
    my_dict = <dictionary described above> 
    keys = key.split(':') 
    if len(keys) == 1: 
     del my_dict[keys[0]] 
    elif len(keys) == 2: 
     del my_dict[keys[0]][keys[1]] 
    elif len(keys) == 3: 
     del my_dict[keys[0]][keys[1]][keys[2]] 
    . . . 

del_key('key_b:key_b_b:key_b_b_a') 
del_key('key_b:key_b_b') 
del_key('key_a') 

我該如何優雅地做到這一點？

Answer 1

它假設您的輸入是有效的密鑰，否則您必須檢查。

data = { 
    "key_a": "value", 
    "key_b": { 
     "key_b_a": "value", 
     "key_b_b": { 
      "key_b_b_a": "value" 
     } 
    } 
} 

def del_key(key): 
    key = key.split(':') 
    temp = data 
    for i in key[:-1]: 
     temp = temp[i] 
    del temp[key[-1]] 
    return data 

print del_key('key_b:key_b_b:key_b_b_a') 
print del_key('key_b:key_b_b') 
print del_key('key_a') 

output: 
{'key_a': 'value', 'key_b': {'key_b_a': 'value', 'key_b_b': {}}} 
{'key_a': 'value', 'key_b': {'key_b_a': 'value'}} 
{'key_b': {'key_b_a': 'value'}}