python将txt文件格式转换成用邻接矩阵表示

Python代码块

import networkx as nx
G = nx.Graph()

path = ‘./edge_list.txt’
edge_list = []
node_set = set() #集合的特性就是元素不会重复，互异的
with open(path, ‘r’) as f:
　　for line in f:
　　　　cols = line.strip().split(’ ')
　　　　y1=int(cols[0])
　　　　y2=int(cols[1])
　　　　node_set.add(y1)
　　　　node_set.add(y2)
　　　　edge = (y1,y2) #元组代表一条边
　　　　edge_list.append(edge)

G.add_nodes_from(range(1,len(node_set)+1)) #节点序号从1开始编号
G.add_edges_from(edge_list)
#A = nx.to_numpy_matrix(G)
A = nx.adjacency_matrix(G).todense()
print(A)

结果显示如下：