Показать список списков в виде дерева в Python

Question

Я бы хотел определить функцию для печати списка списков с общими компонентами в виде дерева.

Например,

lst = [[1, 2, 'a'], [1, 2, 'b'], [1, 1, 'a'], [1, 1, 'b'], [2, 2, 'a']]
build_tree(lst)

должен печатать:

1
    1
        a
        b
    2
        b
2
    2
        a

Я написал следующий код.

def build_tree(lst):
    if len(lst[0]) == 1:
        for e in lst:
            return str(e[0])
    current = lst[0][0]
    seen = []
    tree = ''
    for e in lst:
        if e[0] != current:
            tree += str(current)
            tree += build_tree(seen)
            current = e[0]
            seen = []
        seen.append(e[1:])
    return tree

Но возвращенный результат - чепуха.

Answer 1

Одним из способов решения этой проблемы было бы преобразование вашего списка в структуру дерева, а затем цикл по этому дереву для построения представления строк ваших данных

from collections import defaultdict

tree = lambda: defaultdict(tree)

lst = [[1, 2, 'a'], [1, 2, 'b'], [1, 1, 'a'], [1, 1, 'b'], [2, 2, 'a']]


def make_tree(lst):
    d = tree()    
    for x in lst:
        curr = d
        for item in x:
             curr = curr[item]
    return d

d = make_tree(lst)

def make_strs(d, indent=0):
     strs = []
     for k, v in d.items():
         strs.append('    ' * indent + str(k))
         strs.extend(make_strs(v, indent+1))
     return strs

def print_tree(d):
    print('\n'.join(make_strs(d)))

print_tree(d)

print

1
    2
        a
        b
    1
        a
        b
2
    2
        a