从 csr_matrix 获取节点的邻居

时间:2021-03-18 09:25:42

标签: python matrix indexing graph nodes

我在数据集的实体之间创建了一个稀疏邻接矩阵。我想根据每个节点的邻居计算一些分数。 如果我们假设 adj_mat 是矩阵:

(0, 1)  1.0
(0, 2)  1.0
(1, 0)  1.0
(2, 0)  1.0
(3, 4)  1.0
(4, 3)  1.0
(5, 6)  1.0
(5, 7)  1.0
(6, 5)  1.0
(7, 5)  1.0 

例如,我知道编号为 6 和 7 的节点是节点 5 的邻居。但我不知道如何获得编号 6 和 7。我想要一个循环,为每个节点执行一些操作基于它的邻居。节点的信息已经保存在一个文件中,我们可以通过节点的编号来访问它们。

0 个答案:

没有答案