本文整理汇总了Python中treelib.Tree.nodes方法的典型用法代码示例。如果您正苦于以下问题:Python Tree.nodes方法的具体用法?Python Tree.nodes怎么用?Python Tree.nodes使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类treelib.Tree
的用法示例。
在下文中一共展示了Tree.nodes方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: load
# 需要导入模块: from treelib import Tree [as 别名]
# 或者: from treelib.Tree import nodes [as 别名]
def load(path):
gtype = None
graph = None
nodes = []
#TEMP: LOAD FILE FROM PATH#
file = open(path,"r",0)
#TEMP#
gtype = file.next()
#print isInt(gtype) #ASSERT LINE 0 IS AN INTEGER
gtype = int(gtype)
#print gtype
if (gtype == 0):
graph = Tree()
nodes = []
print 'GRAPH TYPE: TREE'
if (gtype == 1):
graph = digraph()
print 'GRAPH TYPE: EWD'
#print type(graph)
for linenum,line in enumerate(file):
if (linenum == 0):
parts = line.split()
#print parts[0]
nodeAttrs = parts[1].split(';')
numNodeAttrs = nodeAttrs.pop(0)
#print numNodeAttrs
#print isInt(numNodeAttrs)
#print numNodeAttrs, nodeAttrs
continue
if (linenum == 1):
for node in line.split():
tokens = node.split(';')
index = tokens.pop(0)
#print index #in final, assert numNodes == index
#print isInt(index)
if (isInt(index)):
index = int(index)
if (gtype == 0):
nodes.append(Node(identifier=index,data=tokens))
if (index == 0):
#print "ZERO"
#n = nodes[0]
#print type(n)
graph.add_node(nodes[0])
#graph.show()
#print graph.get_node(0)
if (gtype == 1):
graph.add_node(index, attrs=tokens)
#print tokens
continue
if (linenum == 2):
if (gtype == 0):
#print isInt(line) #ASSERT LINE IS A SINGLE INT
numEdges = int(line)
if (gtype == 1):
parts = line.split()
#print parts[0]
lineAttrs = parts[1].split(';')
numLineAttrs = lineAttrs.pop(0)
#print numLineAttrs
#print isInt(numLineAttrs)
#print numLineAttrs, lineAttrs
continue
if (linenum > 2):
#CHECK THAT PARTS ARE INT
parts = line.split()
tail = int(parts[0])
head = int(parts[1])
if (gtype == 0):
graph.add_node(nodes[head],tail)
if (gtype == 1):
attributes = parts[2].split(';')
weight = attributes.pop(0)
#check head and tail are integers
#check number attributes
#print (tail,head), weight, attributes
graph.add_edge((tail,head),weight,attrs=attributes)
if (gtype == 0):
graph.show()
if (gtype == 1):
for node in graph.nodes():
#.........这里部分代码省略.........