python - Python中graphviz中的不同颜色边缘

Question

我正在尝试通过在 Python 中使用 graphviz 来绘制具有不同颜色边缘的图形。我正在创建颜色列表并使用边缘颜色。但是，它看起来在graphviz中不支持。我不确定如何设置不同颜色的边缘。这是我的代码:

import graphviz as gv

d = gv.Digraph()
colors = ['green','red']
d.attr('edge', color = colors)
d.edge('hello','world')
d.edge('world','hello')
d.view()

寻求宝贵意见。谢谢

Answer 1

import graphviz as gv

colors = ['green','red']


def create_graph(colors, d):

  d.edge('hello','world', color=colors[0])
  d.edge('world','hello', color=colors[1])
  d.view()

if __name__ == '__main__':

  d = gv.Digraph()

  create_graph(colors, d)