2015-01-04 83 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()