NetworkX:如何构建 k-Partite Graph 图

NetworkX: How to build a k-Partite Graph graph

我有一组 m 个节点,我们将它们命名为 N0, N1, N2, ..., Nm。使用 networkx 我想构建一个如下图所示的图,即节点排列成 k 层的图,每一层都有 l个节点(每个节点恰好属于1层)。 i 层中的每个节点都与 i+1 层中的每个节点相连(其中 0 <= i <= to k -1).

因此,例如,如果我们有 6 个节点:N0、N1、N2、N3、N4、N5 并且图形应该有 k=3、l=2 那么连接看起来像 N0 连接到 N2, N3; N1连接到N2、N3; N2连接到N4、N5; N3 连接到 N4, N5;

以下应该完成你的工作:

def create_k_partite_graph(layers, node_per_layer):
    g = nx.Graph()
    previous_layer_nodes = []
    actual_layer_nodes = []
    current_node_id = 0
    for k in range(layers):
        for node in range(node_per_layer):
            g.add_node(current_node_id)
            actual_layer_nodes.append(current_node_id)
        
            for previous_node in previous_layer_nodes:
                g.add_edge(current_node_id, previous_node)
                
            current_node_id +=1
        
        previous_layer_nodes = actual_layer_nodes
        actual_layer_nodes = []
    
    return g

一种方法可能如下。

N_LAYERS = 5
N_NODES = 3

g = nx.Graph()

for layer in range(N_LAYERS - 1):
    for u in range(N_NODES):
        for v in range(N_NODES):
            g.add_edge((u,layer),(v,layer+1))

结果: