Как создать список списков для генеалогического древа

Question

Новое для python. Обучение созданию функции, которая приводит к списку списков ... например, что-то, что может представлять семейное дерево. Вот что я придумал:

def familytree(root): 
     many = int(input('How many children does "' + root[0] + '" have? ')) 
     if many > 0: 
      root.append([[] for y in range(many)]) 
      z = root[1] 
      for j in range(many): 
       name = [input("Give name of one of " + root[0] + "'s children?")] 
       z[j] = name 
       familytree(name) 
     print(root) 

Кажется, работает ...

['A', [['B', [['D', [['G'], ['H']]], ['E', [['I']]]]], ['C', [['F', [['J'], ['K']]]]]]] 

... но приводит к ненужным Кронштейны и печатные выходы.

Любые предложения по улучшению реализации?

Будет работать над следующим классом.

Answer 1

Вот как вы можете распечатать окончательный результат. Я также упростил некоторые другие вещи, чтобы сделать его немного яснее.

def familytree(root): 
    many = int(input('How many children does "' + root[0] + '" have? ')) 
    if many > 0: 
     z = [] 
     root.append(z) 
     for j in range(many): 
      name = [input("Give name of one of " + root[0] + "'s children?")] 
      z.append(name) 
      familytree(name) 


root = ['A'] 
familytree(root) 
print(root) 

Answer 2

Один простой подход, не нарушая ваш код:

root = ['A'] 
def familytree(root): 
    many = int(input('How many children does "' + root[0] + '" have? ')) 
    if many > 0: 
     root.extend([] for y in range(many)) 
     for j in root[1:]: 
      name = input("Give name of one of " + root[0] + "'s children? ") 
      j.append(name) 
      familytree(j) 

print(familytree(root)) 

Вы можете также использовать pprint встроенную библиотеку довольно-печать объектов