遞歸在字典

Question

我有一個嵌套的dict它看起來像這樣：

有關鍵children內的多個嵌套。無論何時出現密鑰children，我都希望捕獲密鑰branch。因爲有多個children，我想爲每個孩子做這個。當然，每個孩子也可以有更多的children。這個嵌套可以達到7層。

爲了達到這個目的，我可以寫一個骨頭的7-for循環方法或使用遞歸。所以我給了遞歸了一槍，用下面的代碼上來：

def GatherConcepts(header): 
    if 'children' in header.keys(): 
     if len(header['children']) > 0: 
      if 'branch' in header.keys(): 
       concepts.append(header['handle']) 
       if 'children' in header.keys(): 
        for j in range(0, len(header['children'])): 
         GatherConcepts(header['children'][j]) 
      else: 
       for i in range(0,len(header['children'])): 
        GatherConcepts(header['children'][i]) 

這段代碼的問題是，它給我的只有2級（因爲我調用該函數本身的2倍，因此不使用遞歸正確），而不是7.

我該如何改進以獲得所有級別？

任何指針將不勝感激。

Answer 1

你有一些不必要的冗餘。如果我理解正確，則需要將遞歸與遞歸分開添加到列表中，因爲您想在父項中測試branch。

def GatherConcepts(header): 
    if 'children' in header and 'branch' in header: 
     for child in header['children']: 
      concepts.append(child['handle']) 
      GatherConcepts(child) 

你並不需要測試的header['children']長度 - 如果它是零，那麼循環將只是沒有做任何事情。

Answer 2

爲了正確地得到遞歸，你可以使用這個簡單的模板吧：

def recursive(variable): 
    if something: 
     # base step 
     return somethingelse 
    else: 
     # recursive step 
     return recursive(somethingelse) 

在你的情況，你可以嘗試這樣的事：

def gather_concepts(header): 
    # recursive step 
    if 'branch' in header and 'handle' in header: 
     concepts.append(header['handle']) 
    if 'children' in header: 
     for child in header['children']: 
      return gather_concepts(child) 
    # base step 
    else: 
     return 

你應該調整這個代碼儘管如此，因爲我沒有自己測試過。