收缩 networkx 中的节点列表

Question

我有一个带有节点的字典：

supernodes = list(nx.connected_components(G1))

结果print(supernodes)是：

[{1, 2, 3, 5}, {8, 6}, {7, 9, 10, 12, 13}, {4}, {11}, {14}, {15}]

如何将每个列表合并到一个节点？我找到了这个函数，nx.contracted_nodes(G, (1, 3))但是我如何放置{1,2,3,5}, {8,6}等并创建 7 个签约节点？

Answer 1

你可以试试这个：

import networkx as nx
# Preamble, define G1 

# contract nodes
for supernode in nx.connected_components(G1):
    nodes = sorted(list(supernode))
    for node in nodes[1:]:
        G1 = nx.contracted_nodes(G1, nodes[0], node)

每个节点 x inG1对应于具有 x 作为较小元素的超节点。如果要删除自循环，请改写nx.contracted_nodes(G1, nodes[0], node, self_loops=False).