2017-11-25 189 views
0

如何通過循環添加子節點?我正在使用treeView。我已經定義了行數組。在這個Primary1中,Primary2,Primary3是父節點。 Primary1的Secpndary1子節點。 r [1]定義了Primary或​​Secondary並且r [2]定義了父節點名稱Python:通過循環添加子節點

rows = [('Primary1', 'Primary', None), ('Primary2', 'Primary', None), 
     ('Secpndary1', 'Secondary', 'Primary1'),('Primary3', 'Primary', None)] 


tree = [] 

for r in rows: 
    tree.append({'node_id': r[1], 'children': []}) 

任何人都可以幫助我嗎?

回答

1

試試這個:

rows = [('Primary1', 'Primary', None), ('Primary2', 'Primary', None), 
    ('Secpndary1', 'Secondary', 'Primary1'),('Primary3', 'Primary', None)] 


tree = [] 

for r in rows: 
    if r[2] is None: 
     tree.append({'node_id': r[0], 'children': []}) 
for r in rows: 
    if r[2] is not None: 
     for node in tree: 
      if node['node_id'] == r[2]: 
       node['children'].append({'node_id': r[0], 'children': []}) 
0

這是另一種解決方案,它可以工作。

rows = [('Primary1', 'Primary', None), ('Primary2', 'Primary', None), 
     ('Secpndary1', 'Secondary', 'Primary1'),('Primary3', 'Primary', None), ('Secondary2', 'Secondary', 'Primary1')] 

from collections import defaultdict 
dd = defaultdict(list) 
for row in rows: 
    node , _ , child = row 
    x = dd[child].append(node) if child else dd[node] 

print dd 

>>> dd 
defaultdict(<type 'list'>, {'Primary2': [], 'Primary3': [], 'Primary1': ['Secpndary1', 'Secondary2']})