设置 pyGraphViz 中节点的颜色

Question

网站上提供了这个示例，但没有提到如何使用字典设置颜色。

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

你在使用AGraph初始化后，是否应该使用get_node函数？

- BlueTrin

Answer 1

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

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()

- BlueTrin