在 Networkx 中使用 k-core 获取去除的边

Question

我正在使用 Networkx 的 k-core 删除度数 < 2 的节点。

n=20
G = nx.gnm_random_graph(n=20, m=30, seed=1)
nx.draw(G, with_labels=True)
plt.show()

retain_node_ids = [1, 8]
G.add_edges_from([(u, v) for u in retain_node_ids for v in (n, n+1)])
pos = nx.spring_layout(G)

G = nx.k_core(G, k=2)

nx.draw(G, with_labels=True, pos=pos)
plt.show()

k-core returns 子图。但我也想获得使用 k-core 时删除的节点和边的列表。

任何建议都会非常有用。

Answer 1

nx.k_core只有returns对应的最大子图。为了找到结果子图中未包含的节点和边，您必须找到两个图之间的节点差异，并从那里找到边差异。使用示例 nx.gnm_random_graph:

diff_nodes = set(G.nodes()).difference(H.nodes())
print(diff_nodes)
# {2, 4, 11, 19}

removed_edges = {e for e in G.edges() for n in diff_nodes if n in e}
print(removed_edges)
# {(2, 8), (4, 18), (7, 11), (13, 19)}