python数据结构 树_在Python中打印树数据结构

是的,将__repr__代码移动到__str__,然后在树上调用str()或将其传递给print语句.请记住在递归调用中使用__str__：

class node(object):

def __init__(self, value, children = []):

self.value = value

self.children = children

def __str__(self, level=0):

ret = "\t"*level+repr(self.value)+"\n"

for child in self.children:

ret += child.__str__(level+1)

return ret

def __repr__(self):

return ''

演示：

>>> root = node('grandmother')

>>> root.children = [node('daughter'), node('son')]

>>> root.children[0].children = [node('granddaughter'), node('grandson')]

>>> root.children[1].children = [node('granddaughter'), node('grandson')]

>>> root

>>> str(root)

"'grandmother'\n\t'daughter'\n\t\t'granddaughter'\n\t\t'grandson'\n\t'son'\n\t\t'granddaughter'\n\t\t'grandson'\n"

>>> print root

'grandmother'

'daughter'

'granddaughter'

'grandson'

'son'

'granddaughter'

'grandson'