Преобразовать разделенное пробелами дерево в полезный dict в python - PullRequest
Новая
       3

Преобразовать разделенное пробелами дерево в полезный dict в python

1 голос
/ 24 ноября 2010

У меня есть вывод (список) элементов, таких как: 

Root
  Branch1
    LeafA
    LeafB
  Branch2
    LeafC
      LeafZ
    LeafD

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

Окончательный возможный результат: 

aDict = {
    'Root': null,
    'Branch1': 'Root',
    'LeafA': 'Branch1',
... so on and so forth
}

В конечном счете, я хочу повторитьсловарь и получить ключ и родителя, а также другое значение из другого dict на основе ключа.

Ответы [ 2 ]

4 голосов
/ 24 ноября 2010

Попробуйте это: 

tree = """Root
  Branch1
    LeafA
    LeafB
  Branch2
    LeafC
      LeafZ
    LeafD"""

aDict = {}
iDict = {}
for line in tree.split("\n"):
    key = line.lstrip(" ")
    indent = (len(line) - len(key)) / 2
    if indent == 0:
        aDict[key] = None
    else:
        aDict[key] = iDict[indent - 1]
    iDict[indent] = key

print aDict
# {'LeafD': 'Branch2', 'LeafA': 'Branch1', 'Branch2': 'Root', 'LeafC': 'Branch2', 'LeafB': 'Branch1', 'Branch1': 'Root', 'Root': None, 'LeafZ': 'LeafC'}
0 голосов
/ 26 ноября 2010

Полагаю, это решает проблему: 

#!/usr/bin/env python

def f(txt):
    stack = []
    ret = {}
    for line in txt.split('\n'):
        a = line.split('  ')
        level = len(a) - 1
        key = a[-1]
        stack = stack[:level]
        ret[key] = stack[-1] if len(stack) > 0 else None
        stack.append(key)
    return ret

print f("""Root
  Branch1
    LeafA
    LeafB
  Branch2
    LeafC
      LeafZ
    LeafD""")

print f("""Root1
  Branch1
    LeafA
      LeftZ
  Branch2
    LeftB
Root2
  Branch3""")

Вывод: 

{'LeafD': 'Branch2', 'LeafA': 'Branch1', 'Branch2': 'Root', 'LeafC': 'Branch2', 'LeafB': 'Branch1', 'Branch1': 'Root', 'Root': None, 'LeafZ': 'LeafC'}
{'LeafA': 'Branch1', 'Branch2': 'Root1', 'Branch1': 'Root1', 'LeftZ': 'LeafA', 'LeftB': 'Branch2', 'Branch3': 'Root2', 'Root1': None, 'Root2': None}
...