python - 如何使用 PyGraphviz 在无向图的边上添加和显示权重？

Question

import pygraphviz as pgv
A = pgv.AGraph()
A.add_node('Alice')
A.add_node('Emma')
A.add_node('John')
A.add_edge('Alice', 'Emma')
A.add_edge('Alice', 'John')
A.add_edge('Emma', 'John')
print A.string()
print "Wrote simple.dot"
A.write('simple.dot')  # write to simple.dot
B = pgv.AGraph('simple.dot')  # create a new graph from file
B.layout()  # layout with default (neato)
B.draw('simple.png')  # draw png
print 'Wrote simple.png'

我想为边缘添加权重，这些边缘也应该显示在图中。

Answer 1

您可以在创建边缘时向边缘添加属性:

A.add_edge('Alice', 'Emma', weight=5)

或者您可以稍后设置它们:

edge = A.get_edge('Alice', 'Emma')
edge.attr['weight'] = 5

要向边缘添加文本信息，请改为给它们一个 label 属性:

edge = A.get_edge('Alice', 'Emma')
edge.attr['label'] = '5'

所有属性都在内部存储为字符串，但 GraphViz 将这些解释为特定类型；查看attribute documentation .