如何修复此方法以返回节点在位置 i 与其他节点建立的所有连接

时间:2021-06-14 13:51:43

标签: python networkx

给定一个有 2 个节点 (0, 1) 的有向图:0 连接到 0 和 1。1 连接到 0。就像这个邻接矩阵:

Adj = [[1,1], # 1 represents connections and 0 no connection
       [1,0]]

我想要的输出如下:

v = [{0,1}, {0}]  # Node 0 connects to itself and 1. Node 1 connects to node 0.

这意味着,位于 v[0] 的节点 0 连接到节点 0 和 1。位于 v[1] 的节点 1 仅连接到节点 0。所以,基本上它应该告诉与位置 i 的节点相连的节点

我做了什么:

>> import networkx as nx
>> import numpy as np
>> arr = np.array([[1,1],[1,0]])
>> G = nx.DiGraph()
>> G = nx.from_numpy_array(arr, create_using = nx.DiGraph)
>> L = []
>> for i in range (N): # N is the amount of nodes in the graph, in this case 2.
>>    aux = set(nx.edge_bfs(nx.Graph(G.edges), i))
>>    U.append(aux)
>> print(U)

Output: [{(0, 1), (0, 0)}, {(1, 0), (0, 0)}]

1 个答案:

答案 0 :(得分:1)

我找到的解决方案是使用 nx.descendants 方法。

    for i in range (0, q):
      x = nx.descendants(G,i)

该代码段给出了从源节点可到达的所有节点,自循环除外。