将空格分隔的树转换为Python中的有用字典

Question

我有一个输出（一个列表），像这样：

Root
  Branch1
    LeafA
    LeafB
  Branch2
    LeafC
      LeafZ
    LeafD

这些项之间都是用两个空格分开的。 我想要构建一个这个列表的逻辑表示，去掉开头的空格，同时保留父子关系。

最终可能的结果是：

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

最后，我想要遍历这个字典，获取键（Key）和父项，以及根据键从另一个字典中获取的其他值。

Answer 1

我想这段代码解决了这个问题：

#!/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}

Answer 2

试试这个：

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'}