【Python】networkx中如何获取图的邻接矩阵

需要调取networkx中graph实例的邻接矩阵,搜“network 邻接矩阵”没有迅速找到解决方案。我写一个吧。

错误获取:

networkx中有个adjacency_matrix()函数,得到的邻接表看形状虽然是N*N,但是打印出来会发现是这个格式:

  (0, 1)    1
  (0, 30)    1
  (0, 33)    1
  (0, 99)    1

第一列是源头和终止节点,第二列是权重。

但是,如果你下列语句遍历,是会报错的:ValueError: not enough values to unpack (expected 2, got 1)

# 错误示范

for (x,y), w in nx.adjacency_matrix(G):

    print(x, y, w)

正确获取:

import networkx as nx
import numpy as np
A=np.array(nx.adjacency_matrix(G).todense())

转换为numpy矩阵是因为原始的格式不支持A[i][j]形式的索引。 

成功得到邻接矩阵,解决!

[[0 0 0 ... 0 0 1]
 [0 0 1 ... 0 0 0]
 [0 1 0 ... 0 0 0]
 ...
 [0 0 0 ... 0 0 0]
 [0 0 0 ... 0 0 1]
 [1 0 0 ... 0 1 0]]

 

你可能感兴趣的:(Python编程手册)