我有一个无向 NetworkX 图，想找到 node_a 的所有邻居，它们也是 node_b 的邻居

Question

直截了当的问题：如何在 node_a 和 node_b 之间找到所有共同的邻居，最好的方法是什么？

我的代码是：

my_node_list_a = [ ... ]
my_node_list_b = [ ... ]

for c in my_node_list_a:
    for x in nx.neighbors(G, c):
        for y in my_node_list_b:
            for h in nx.neighbors(G, y):
                 if x in y:
                    print(x)

这是错误的代码。首先，我必须通过执行for i in G.nodes(): print(i) 手动填充my_node_list_a 和my_node_list_b，并且只将我想要运行循环的节点放入数组中。

编辑：

标题说的是邻居，但我是 networkx 的新手，所以它可能不是正确的术语。如果邻居只表示距离为 1 的节点，那么是的，这就是我想要的。

但是，我也对哪些节点连接数组 my_node_list_a 和 my_node_list_b 中距离大于 1 的节点感兴趣，所以如果有人能告诉我如何做到这一点，它也会很有用（基本上，向我展示它们之间的路径）

进一步编辑：

我需要根据所选节点的邻居从我的主图 G 创建一个图。

最好的方法是什么？

Answer 1

要查找两个节点之间的共同邻居，您可以使用函数common_neighbors()：

import networkx as nx

G = nx.turan_graph(6, 2)

%matplotlib inline # jupyter notebook
nx.draw(G, with_labels=True)

list(nx.common_neighbors(G, 3, 5))
# [0, 1, 2]

要查找两组节点（顶点）之间的共同邻居，您可以使用函数node_boundary()：

import networkx as nx
from itertools import chain

G = nx.grid_graph(dim=[5, 5])

list_A = [(3, i) for i in range(5)]
list_B = [(1, i) for i in range(5)]
intersec = nx.node_boundary(G, list_A) & nx.node_boundary(G, list_B)
# {(2, 1), (2, 0), (2, 3), (2, 2), (2, 4)}

color = []
for i in G.nodes():
    if i in intersec:
        col = 'green'
    elif i in chain(list_A, list_B):
        col = 'red'
    else:
        col = 'pink'
    color.append(col)

%matplotlib inline
nx.draw(G, with_labels=True, node_color=color, node_size=1000)

Answer 2

你可以这样做： （如果你有一个大图表，我改变了我的答案，以避免你经历大量）

import networkx as nx

def keep_neighbors(graph, node):
    tokeep = set()
    for k, v in g[node].items():
        # print v.get('weight') == 1
        if v.get('weight') == 1:
            tokeep.add(k)
    return tokeep

if __name__ == '__main__':
    g = nx.Graph()
    g.add_edge('a', 'b', weight = 1)
    g.add_edge('a', 'c', weight = 1)
    g.add_edge('a', 'd', weight = 2)
    g.add_edge('a', 'e', weight = 1)
    g.add_edge('a', 'f', weight = 1)

    g.add_edge('z', 'c', weight = 1)
    g.add_edge('z', 'd', weight = 1)
    g.add_edge('z', 'e', weight = 1)

    g.add_edge('k', 'c', weight = 1)
    g.add_edge('k', 'd', weight = 1)
    g.add_edge('k', 'e', weight = 1)


    a = keep_neighbors(g, 'a')
    b = keep_neighbors(g, 'z')
    print(a) # set(['c', 'b', 'e', 'f'])
    print(b) # set(['c', 'e', 'd'])

    result = a & b
    print(result) # set(['c', 'e'])

    a = keep_neighbors(g, 'k')
    b = keep_neighbors(g, 'z')
    result = a & b
    print(result) # set(['c', 'e', 'd'])