如何在NetworkX中获取指定节点的关联节点与边并生成子图？

我用Python的NetworkX库创建了一个简单无向图：

import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()
G.add_nodes_from(["A", "B", "C", "D", "E"])
G.add_edges_from([("A","C"), ("B","D"), ("B","E"), ("C", "E"), ("A", "E"), ("E", "D")])
#pos = nx.random_layout(G)

fig, ax = plt.subplots()
nx.draw_networkx(G, pos=pos, ax=ax)
fig.tight_layout()
plt.show()

我希望返回与指定节点相连的所有节点和边组成的子图。比如指定节点"A"时，关联节点是"C"和"E"，需要返回这3个节点及它们之间的边。

我尝试了以下代码，但返回的节点数量超出预期：

node = "A"
# connected_nodes = nx.shortest_path(G,node).keys()
connected_nodes = list(nx.node_connected_component(G, "A"))
connected_edges = G.edges([node])
print(connected_nodes)
print(connected_edges)

H = nx.Graph()
H.add_nodes_from(connected_nodes)
H.add_edges_from(connected_edges)

fig, ax = plt.subplots()
nx.draw_networkx(H, pos=pos, ax=ax)
fig.tight_layout()
plt.show()

问题出在nx.node_connected_component(G, "A")返回的是整个连通分量的所有节点（你的图是全连通的，所以会返回全部5个节点），而你需要的是指定节点本身加上它的直接邻居节点，再提取这些节点之间的所有边。

正确的实现步骤：

• 获取目标节点集合：指定节点 + 它的所有直接邻居
• 用nx.induced_subgraph提取这些节点构成的诱导子图（自动包含这些节点之间的所有边）

修改后的代码：

import networkx as nx
import matplotlib.pyplot as plt

# 原始图创建
G = nx.Graph()
G.add_nodes_from(["A", "B", "C", "D", "E"])
G.add_edges_from([("A","C"), ("B","D"), ("B","E"), ("C", "E"), ("A", "E"), ("E", "D")])
pos = nx.random_layout(G)  # 取消注释，确保布局变量有定义

# 指定目标节点
target_node = "A"

# 获取目标节点集合：自身 + 直接邻居
target_nodes = [target_node] + list(G.neighbors(target_node))

# 提取诱导子图（自动包含这些节点之间的所有边）
H = nx.induced_subgraph(G, target_nodes)

# 绘制子图
fig, ax = plt.subplots()
nx.draw_networkx(H, pos=pos, ax=ax)
fig.tight_layout()
plt.show()

# 验证输出
print("目标节点集合：", target_nodes)
print("子图的边：", list(H.edges()))

代码说明 #

• G.neighbors(target_node)：返回指定节点的所有直接邻居节点（无向图中是双向关联）
• nx.induced_subgraph(G, target_nodes)：生成由指定节点集合构成的子图，自动保留这些节点在原图中的所有边（比如A、C、E之间的(A,C)、(A,E)、(C,E)都会被包含）
• 复用原始图的布局pos，可以让子图节点位置和原图一致，更直观对比

内容的提问来源于stack exchange，提问作者G. Macia