NetworkX：ノードの第2レベルを見つける

Question

私は有向グラフGを持っています。私はUserノードをnレベルに返すことができる関数を作ろうとしています。 nx.bfs_successors(G,node)戻るノードの多数のメモリを占有することができるリストとして後継の全てを使用

def findNodes(node, level): 
return #nodes at this level 

。

Answer 1

あなたはnx.ego_graphについて知っていますか？

def find_nodes(graph, node, distance): 

    # get all nodes within distance around the query node 
    nodes = set(nx.ego_graph(graph, node, radius=distance)) 

    # remove nodes that are not **at** the specified distance but closer 
    if distance > 1: 
     nodes -= set(nx.ego_graph(graph, node, radius=distance-1)) 

    return list(nodes)