我正在使用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。
还在线检查了其他帖子,但找不到信息。
答案 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
跟踪最大深度(或嵌套深度)。要计算每个节点的最大深度,可以使用一个循环(遍历所有节点)。