NetworkX从邻接矩阵创建图时如何避免自动添加边权重
nx.from_numpy_array 默认会把邻接矩阵的非零值写入边的weight属性,没有原生的一键开关可以直接禁用这个行为,但是可以通过以下两种方案在创建阶段就避免该属性生成,不需要事后清理:
方案1:高版本直接指定参数(NetworkX 2.7+)
2.7及以上版本的from_numpy_array提供了edge_attr控制参数,显式传入None即可阻止自动添加权重属性,生成的边不会携带任何默认属性:
G = nx.from_numpy_array( A, create_using=nx.DiGraph, edge_attr=None )
方案2:全版本兼容实现
如果使用低于2.7的版本,可以直接提取邻接矩阵的边坐标,通过边列表创建图,整个过程不会自动生成任何边属性,图结构和原方法生成的完全一致:
import numpy as np # 提取有向边:矩阵行索引对应源节点,列索引对应目标节点 edges = np.argwhere(A == 1) G = nx.DiGraph() G.add_edges_from(edges)
注:默认添加
weight属性是该函数的历史设计逻辑,主要是为了适配最短路径、中心性计算等默认读取weight属性的算法,对于无权0-1邻接矩阵,上述两种方法都能从根源避免多余属性生成。
内容的提问来源于stack exchange,提问作者Archie Gertsman
相关产品推荐
相关产品推荐

