networkx是否提供最大深度(或嵌套深度)?

时间:2018-10-25 21:04:04

标签: python networkx graph-theory depth-first-search depth

我正在使用networkx构建项目图。对于特定的图,我需要每个节点的最大深度(或嵌套深度)(类似于this)。

例如。

我的图形中有多个节点,例如-

G -> d, foo, bar, tar, zar, car, char, jar, par, radar, far, ....

d与其他类似的人

d -> {'foo': {'bar': {'tar': 2}, 'zar': {'car': {'char': 1}, 'jar': 'par'}}, 'radar': 'far'}

我想要节点的最大深度(或连通性)。

因此,在这种情况下- d->foo->zar->car->char(总共5个节点)

有什么方法可以使用networkx来计算(我有超过1M个节点,因此数据量很大!)?

我查看了他们的手册here

还在线检查了其他帖子,但找不到信息。

1 个答案:

答案 0 :(得分:0)

不,他们没有。

因此,我编辑了他们的github代码以添加此功能。

代码:

max_d = []

#dfs_depth base Code reference networkx

def dfs_depth(G, source=None, depth_limit=None): if source is None: nodes = G else: nodes = [source] visited = set() if depth_limit is None: depth_limit = len(G) for start in nodes: print(start) if start in visited: continue max_depth = 0 visited.add(start) stack = [(start, depth_limit, iter(G[start]))] while stack: parent, depth_now, children = stack[-1] try: child = next(children) if child not in visited: yield parent, child visited.add(child) if depth_now > 1: if((depth_limit - depth_now + 1)>max_depth): max_depth = depth_limit - depth_now + 1 stack.append((child, depth_now - 1, iter(G[child]))) except StopIteration: stack.pop() global max_d max_d.append(max_depth)

此处max_d跟踪最大深度(或嵌套深度)。要计算每个节点的最大深度,可以使用一个循环(遍历所有节点)。