python - Graphviz:为具有彩虹效果的线条着色

Question

我有数据框，我使用 graphviz 构建图形

for id_key, group in df.groupby('ID'):
f = Digraph('finite_state_machine', filename='fsm.gv', encoding='utf-8')
f.body.extend(['rankdir=LR', 'size="5,5"'])
f.attr('node', shape='box')
for i in range(len(group)-1):
    f.edge(str(group['category'].iloc[i]), str(group['category'].iloc[i+1]),
           label=str(group['search_term'].iloc[i+1]))
f.render(filename=str(id_key))

得到这个结果 .如何更改线条颜色:第一个箭头 - 红色，第二个 - 橙色，第三个 - 黄色等？

Answer 1

您可以使用 brewer color schemes 之一.例如:

g = graphviz.Digraph(format='png')
g.body.extend(["rankdir=LR"])
for i in range(9):
    g.edge(str(i),str(i+1),color="/spectral9/"+str(i+1))
g.render(filename="example")

生产:

如果您想自己生成颜色，可以使用 hsv format具有常量饱和度和值并增加色调:

n = 20
g = graphviz.Digraph(format='png')
g.body.extend(["layout=circo"])
for i in range(n):
    g.edge(str(i),str(i+1),color="{h:} 1 1".format(h=i/n))
g.edge(str(n),str(0),color="1 1 1")
g.render(filename="example")

生产: