在pyGraphViz中设置节点的颜色

时间:2015-01-04 00:03:25

标签: python python-2.7 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 个答案:

答案 0 :(得分:4)

发现你应该有效地使用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()