Блог

У меня есть набор из m узлов, назовем их N0, N1, N2, …, Nm. Используя, networkx я хотел бы построить график, подобный тому, что изображен на рисунке ниже, т. Е. График, в котором узлы расположены в k слоев, каждый слой имеет l узлов (каждый узелпринадлежит ровно 1 слою). Каждый узел из слоя i связан с каждым узлом в слое i 1 (где 0 <= i <= к 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;

Подход может быть следующим.

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

Результат:

Следующее должно выполнить вашу работу:

 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