python networkx如何抽取子图_【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]]