首页 > 解决方案 > 从 csr_matrix 获取节点的邻居

问题描述

我在数据集的实体之间创建了一个稀疏邻接矩阵。我想根据每个节点的邻居计算一些分数。如果我们假设 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。我想要一个循环,根据其对每个节点执行一些操作邻居。节点的信息已经保存在一个文件中,我们可以通过节点的编号来访问它们。

标签: pythonmatrixindexinggraphnodes

解决方案


推荐阅读