Networkx中边缘宽度生效但Pyvis中不显示的问题

问题背景 #

基于Pandas DataFrame构建的Networkx图，边缘权重可在Networkx中正常渲染为不同宽度的边，但通过Pyvis可视化时，边宽始终无法根据权重变化。尝试循环调用add_edge设置value属性也未生效。

原因分析 #

1. 属性映射不匹配：Networkx图中的edge_weight属性不会自动映射到Pyvis控制边宽的value属性，net.from_nx(G)导入时只会保留原始属性名，Pyvis无法识别edge_weight作为边宽依据。
2. 重复添加边无效：调用net.from_nx(G)后再用add_edge添加同名边，会创建重复边而非修改已有边的属性，导致权重设置无法覆盖原有边的默认配置。

解决方案 #

方案1：导入后修改Pyvis边的value属性

在通过from_nx导入图后，直接遍历Pyvis的边对象，将Networkx中的edge_weight赋值给Pyvis边的value属性：

# Graph visualization using Pyvis
net = Network(width='1900px', height='900px', bgcolor='#222222', font_color='white')
net.repulsion()
# 从Networkx导入图结构
net.from_nx(G)

# 遍历所有边，设置value属性（控制边宽）
for edge in net.edges:
    # 从Networkx图中获取对应边的权重
    edge_weight = G.edges[edge['from'], edge['to']]['edge_weight']
    edge['value'] = edge_weight

net.show('test.html')

方案2：构建Networkx图时直接使用value作为边属性名

在创建Networkx图时，将DataFrame中的edge_weight映射为value属性，这样from_nx导入时Pyvis会自动识别该属性并应用为边宽：

# generate a Networkx graph，将edge_weight映射为value属性
G = nx.from_pandas_edgelist(
    df, 
    source='node1', 
    target='node2', 
    edge_attr={'value': 'edge_weight'},  # 直接指定属性映射
    create_using=nx.Graph()
)

# 节点属性设置代码保持不变
node_size = {}
for index, row in df.iterrows():
    node_size[row['node1']] = row['node1_size']
nx.set_node_attributes(G, node_size, 'size')

# Graph visualization using Pyvis
net = Network(width='1900px', height='900px', bgcolor='#222222', font_color='white')
net.repulsion()
# 导入后直接生效，无需额外设置
net.from_nx(G)
net.show('test.html')

补充说明 #

Pyvis中边的宽度由value属性控制，数值越大边越宽。如果权重数值过小（如示例中的0.5、0.6），可以考虑乘以一个缩放系数（比如edge['value'] = edge_weight * 10）来让宽度差异更明显。

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