2015-01-04 99 views
0

本示例在网站上给出,但没有提及如何使用字典设置颜色。在pyGraphViz中设置节点的颜色

import pygraphviz as pgv 
d={'1': {'2': None}, '2': {'1': None, '3': None}, '3': {'2': None}} 
A=pgv.AGraph(d) 

你是否应该在通过AGraph初始化后使用get_node?

回答

1

发现您应该使用有效的get_node()并传递id来查找节点。还发现你需要设置图形的样式来填充,否则你将看不到填充颜色。

import pygraphviz as pgv 
from PIL import Image 

d={'1': {'2': None}, '2': {'1': None, '3': None}, '3': {'2': None}} 
A=pgv.AGraph(d) 
A.node_attr['style']='filled' 
n = A.get_node('1') 
n.attr['fillcolor']="#CCCCFF" 
n.attr['label'] = 'MY LOVELY LABEL' 
A.layout() # default to neato 
A.layout(prog='dot') # use dot 
A.draw('file.png') 
img = Image.open('file.png') 
img.show()