python networkx- enregistrer des graphiques déconnectés d'un seul chiffre dans des fichiers séparés

Question

J'ai un programme qui génère mon graphique et l'affiche sur une seule figure.

Edges2 = [(1, 2), (1, 3), (1, 4), (4, 5), (6, 7), (6,8)]

G = nx.DiGraph()

la fonction qui génère le graphe est le suivant:

def create_graph(G,nodes,Sets): 

G.add_edges_from(nodes) 

#value assigned to each world 
custom_labels={} 
custom_node_sizes={} 
node_colours=['y'] 

for i in range(0, len(Sets)): 
    custom_labels[i+1] = Sets[i] 
    custom_node_sizes[i+1] = 5000 
    if i < len(Sets): 
     node_colours.append('b') 



nx.draw(G,labels=custom_labels,node_list = nodes,node_color=node_colours, node_size=custom_node_sizes.values()) 
#show with custom labels 
plt.show() 

à la fonction ci-dessus je passe la liste des arêtes (Edges2). La fonction génère deux graphiques déconnectés sur une seule figure. Cependant, j'aimerais sauvegarder ces deux graphiques séparément.

Donc, fondamentalement, existe-t-il un moyen de sauvegarder deux graphiques déconnectés dans deux fichiers? Donc, je peux obtenir graph1.png et graph2.png.

Answer 1

Je n'ai pas tout à fait obtenu l'entrée 'Sets' à votre fonction ou pourquoi add_edges_from(nodes) vous utilisez des nœuds en entrée et non des bords !. Donc, pour répondre à votre question de tracer des graphes déconnectés dans 2 fichiers séparés, j'ai reproduit le problème sans custom_labels car cela dépend de l'entrée 'Sets' et j'ai envoyé le Edges2 comme une entrée pour les nœuds et les Sets aussi. Comme suggéré par @joel j'ai utilisé la fonction weakly_connected_component_subgraphs, puis bouclé la sortie de la fonction enregistrer chaque graphique séparément. Donc, à la fin, le graphique original est enregistré dans original_graph.png, les sous-graphiques sont enregistrés dans graph1.png et graph2.png respectivement.

def create_graph(G,nodes,Sets): 
    G.add_edges_from(nodes) 

    #value assigned to each world 
    custom_labels={} 
    custom_node_sizes={} 
    node_colours=['y'] 

    for i in range(0, len(Sets)): 
     custom_labels[i+1] = Sets[i] 
     custom_node_sizes[i+1] = 5000 
     if i < len(Sets): 
      node_colours.append('b') 
    nx.draw(G,node_list = nodes,node_color=node_colours, node_size=1000, with_labels = True) 
    plt.savefig("original_graph.png") 
    plt.show() 
    G_comp = nx.weakly_connected_component_subgraphs(G) 
    i = 1 
    for comp in G_comp: 
     nx.draw(comp,node_color=node_colours, node_size=1000, with_labels=True) 
     #show with custom labels 
     fig_name = "graph" + str(i) + ".png" 
     plt.savefig(fig_name) 
     plt.show() 
     i += 1 


Edges2 = [(1, 2), (1, 3), (1, 4), (4, 5), (6, 7), (6,8)] 
G = nx.DiGraph() 
create_graph(G,Edges2,Edges2) 

graphique originale

graph1 et graph2

SOUS LA DIRECTION DE AKI: J'ai ajouté les étiquettes que je avais besoin (voir commentaires). La dernière partie du code est:

i = 1 
    custom_number = 1; 
    for comp in G_comp: 
     dictfilt = lambda x, y: dict([ (i,x[i]) for i in x if i in set(y) ]) 
     wanted_keys = (range(custom_number,custom_number + len(comp))) 
     newdict = dictfilt(custom_labels, wanted_keys) 

     nx.draw(comp,node_color=node_colours, node_size=1000, with_labels=True, labels = newdict) 
     #show with custom labels 
     fig_name = "graph" + str(i) + ".png" 
     plt.savefig(fig_name) 
     plt.show() 
     custom_number += len(comp) 
     i += 1 

Cette version améliorée récupère les données nécessaires du dictionnaire. Un grand merci à @author de la réponse

Answer 2

depuis votre graphique est dirigé, vous devez extraire composants fortement connectés de votre graphique:

graphs = nx.strongly_connected_component_subgraphs(G) 

plus de détails peuvent être trouvés here
il y a d'autres méthodes utiles pour les composants qui vous pouvez trouver here