GeezerVison is showing github.com — algorithms/algorithms/tree/pretty_print.py at master · oldmonad/algorithms

Expand file tree

pretty_print.py

Copy path

More file actions

23 lines (20 loc) · 766 Bytes

Copy path

23 lines (20 loc) · 766 Bytes

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

# a -> Adam -> Book -> 4

# b -> Bill -> Computer -> 5

# -> TV -> 6

# Jill -> Sports -> 1

# c -> Bill -> Sports -> 3

# d -> Adam -> Computer -> 3

# Quin -> Computer -> 3

# e -> Quin -> Book -> 5

# -> TV -> 2

# f -> Adam -> Computer -> 7

from __future__ import print_function

def tree_print(tree):

for key in tree:

print(key, end=' ') # end=' ' prevents a newline character

tree_element = tree[key] # multiple lookups is expensive, even amortized O(1)!

for subElem in tree_element:

print(" -> ", subElem, end=' ')

if type(subElem) != str: # OP wants indenting after digits

print("\n ") # newline and a space to match indenting

print() # forces a newline