时间戳

首页 > 解决方案 > 如何使用Python获取存储在字典中的边列表中节点的三角形计数和局部聚类系数？

python - 如何使用Python获取存储在字典中的边列表中节点的三角形计数和局部聚类系数？

问题描述

我有一本字典，其中包含许多具有以下格式的边缘列表：

{'0': [[0, 2],
  [1, 2],
  [3, 4],
  [5, 6]],
 '1': [[1, 6],
  [2, 6],
  [3, 4],
  [0, 3],
  [2, 4]]}

我想首先获得每个边列表 0 和 1 的三角形总数。

然后，对于边缘列表 0 和 1 中的每个节点，我想获得每个节点所属的三角形的数量及其局部聚类系数。

结果格式可以是字典、列表或熊猫数据框。

提前致谢！

标签： pythongraphnetworkx

解决方案

在您提供的数据中，您的数据实际上没有任何三角形（三个节点上的完整子图），但这是一个带有附加边列表的示例。请注意，该networkx.from_edgelist函数接受一个边缘列表作为元组列表而不是列表列表，因此我不得不更改它们。该networkx.triangles函数返回一个由每个节点键入的字典，该字典提供使用该节点的三角形的计数。我们可以将该字典的值相加并除以三以确定三角形的实际数量，请参见文档。

import networkx as nx
dat = {'0': [[0, 2],[1, 2],
             [3, 4],[5, 6]],
       '1': [[1, 6],[2, 6],
             [3, 4],[0, 3],[2, 4]],
       # adding a graph with two triangles
       '2': [[0,1],[1,2],
             [2,0],[3,4],
             [4,5],[5,3]]}

graphs = {key:nx.from_edgelist([(v[0],v[1]) for v in value]) for key,value in dat.items()}

# count number of triangles per node
node_triangle_counts = {key:nx.triangles(graph) for key,graph in graphs.items()}
print(node_triangle_counts)
# {'0': {0: 0, 2: 0, 1: 0, 3: 0, 4: 0, 5: 0, 6: 0},
#  '1': {1: 0, 6: 0, 2: 0, 3: 0, 4: 0, 0: 0},
#  '2': {0: 1, 1: 1, 2: 1, 3: 1, 4: 1, 5: 1}}

# sum the node counts and divide by three because each triangle is counted for all three nodes in the triangle
num_triangles_per_graph = {key:sum(val_dict.values()) / 3
                           for key,val_dict in node_triangle_counts.items()}

# {'0': 0, '1': 0, '2': 2}

推荐阅读