Python NetworkX：如何构建k部图

我有一组m个节点，让我们把它们命名为N0，N1，N2，…，Nm。使用

networkx

我想构建一个如下图所示的图，即一个图，其中节点排列为k层，每层有l个节点（每个节点恰好属于1层）。层i中的每个节点都与层i+1中的每个节点连接（其中0以下内容应完成您的工作：

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))

结果: