python - Networkx 常见邻居未正确过滤，未返回正确的邻居列表

Question

我有一个要过滤的测试图

import networkx as nx
import pickle
from pyvis.network import Network

central_node='a'

nono=['a', 'b', 'c', 'd', 'e', 'f', 'g','h','i','j','k']
eded=[('a', 'b'), ('a', 'c'), ('a', 'd'),('a','f'),('b', 'e'),('c', 'e'),('f', 'd'),('a', 'g'),('e','h'),('e','g'),('i','e'),('j','f'),('k','j')]

graph=nx.Graph()
graph.add_nodes_from(nono)
graph.add_edges_from(eded)

net=Network(height='100%', width='100%', bgcolor='white', font_color='black')
net.from_nx(graph)
net.show('test Graph.html')

这给出了这张图

全图

然后，我将过滤掉中心节点“a”周围半径 2 以外的所有节点

subgraph=nx.ego_graph(G=graph,n=central_node, radius=2, center=True)
subnet.from_nx(subgraph)
subnet.show('test SubGraph.html')

这给出了这张图（到目前为止一切正常）

子图

现在，困难的部分。我试图保留与我的中心节点“a”至少有 3 个节点的子图中的节点（在我的情况下，节点“e”是我想要保留的唯一一个）。

for node in subgraph.nodes:
    neighbor = nx.common_neighbors(subgraph,central_node,node)
    n=len(list(neighbor))
    print(node,n)
    print(list(neighbor))
    for i in sorted(neighbor):
        filtered_nodes.extend(i)
    if n > 2:
        neighbors[node]=n
        filtered_nodes.append(node)
        filtered_nodes.extend(sorted(neighbor))

并想显示结果图（显示节点 a、b、c、e、g 及其连接边）

但这是结果

filtered_graph=subgraph.subgraph(filtered_nodes)
final_net=Network(height='100%', width='100%', bgcolor='white', font_color='black')
final_net.from_nx(filtered_graph)
final_net.show('Traffic Graph.html')

最终图表

似乎 common_neighbors 函数确实返回了节点 b、c、g 的列表。每当我尝试打印它时，它都会显示一个空列表。另一方面，它可以正确计算节点和中心节点“a”之间的公共邻居的数量。

a 5
[]
b 0
[]
c 0
[]
d 1
[]
e 3
[]
f 1
[]
g 0
[]
j 1
[]

我无法正确弄清楚我做错了哪一部分。

标签： pythonnetworkxpyvis

Answer 1

common_neighbors 函数实际上并不返回一个列表——相反，它返回一个迭代器，它可以用来生成一个列表。结果，当您这样做时n=len(list(neighbor))，您实际上使用了迭代器，将其留空。要解决此问题，请在其返回值上调用 list 并将其保存到变量中：

for node in subgraph.nodes:
    neighbor = list(nx.common_neighbors(subgraph, central_node, node))
    n = len(neighbor)
    print(node, n)
    print(neighbor)
    for i in sorted(neighbor):
        filtered_nodes.extend(i)
    if n > 2:
        neighbors[node] = n
        filtered_nodes.append(node)
        filtered_nodes.extend(sorted(neighbor))