如何用snap.py绘制可按权重调整边宽的带权图？

没问题，我来帮你解决这个问题。SNAP确实提供了带权图的实现类，而且可以通过Graphviz的属性来控制边的粗细，完全满足你的需求，具体步骤如下：

1. 替换无权图类为带权无向网络类 #

你之前用的PUNGraph是无权无向图，要存储边的权重，需要换成SNAP的TUNet类——这是专门用于无向带权网络的结构，每个边可以关联一个浮点型的权重值。初始化代码改为：

import snap

# 创建无向带权网络
G = snap.TUNet.New()

2. 添加带权重的边 #

使用AddEdge方法时，第三个参数就是边的权重值，示例如下：

# 添加节点
G.AddNode(1)
G.AddNode(2)
G.AddNode(3)
G.AddNode(4)

# 添加带权重的边
G.AddEdge(1, 2, 2.0)   # 权重2.0
G.AddEdge(2, 3, 5.0)   # 权重5.0（这条边会更粗）
G.AddEdge(3, 4, 1.5)   # 权重1.5
G.AddEdge(1, 4, 3.0)   # 权重3.0

3. 自定义绘图逻辑，按权重调整边粗细 #

SNAP的DrawGViz函数支持通过回调函数自定义节点和边的Graphviz属性。我们可以给每条边设置penwidth（线条宽度）属性，值由权重缩放而来（比如乘以一个系数让粗细差异更明显）。

完整的绘图代码示例：

def set_edge_attrs(Graph, Edge, Attrs):
    # 获取当前边的权重
    weight = Graph.GetEDat(Edge.GetId())
    # 设置penwidth，这里乘以0.8是为了调整粗细的缩放比例，可根据需求修改
    Attrs.AddAttr("penwidth", str(weight * 0.8))
    # 如果需要同时显示权重标签，可以加上这行（可选）
    # Attrs.AddAttr("label", str(weight))

# 调用DrawGViz绘制图，指定边属性回调函数
snap.DrawGViz(G, snap.gvlDot, "weighted_graph.png", "Weighted Graph", edgeAttrs=set_edge_attrs)

关键说明 #

• TUNet是TNodeEDatNet<TInt, TFlt>的别名，意味着节点ID是整数类型，边的权重是浮点类型；如果需要其他类型的权重，也可以自定义TNodeEDatNet的模板参数。
• penwidth的缩放系数可以根据你的权重范围调整：如果权重值普遍偏大，就用更小的系数；如果权重偏小，就用更大的系数，保证边的粗细差异直观。
• 如果你需要绘制有向带权图，可以换成TNGraph对应的带权类TNet（有向带权网络），用法类似。

这样绘制出来的图，边的粗细会直接对应权重大小，完全符合你的需求～

内容的提问来源于stack exchange，提问作者Jiho Choi