Python: networkx customize graph node labels?

Question

I have the following example code:

import pandas as pd
import networkx as nx

import matplotlib.pyplot as plt


def view_graph(graph, labels):
    '''
        Plots the graph
    '''
    pos = nx.spring_layout(graph)

    nx.draw(graph, pos, with_labels=True)
    nx.draw_networkx_edge_labels(graph, pos,edge_labels=labels)
        
    plt.axis('off')
    plt.show()

index_to_name = {1: "Paul", 2: "Magda", 3: "Paul", 4: "Anna", 5: "Marie", 6: "John", 7: "Mark"}
relation = {}
g = nx.DiGraph()

g.add_edge(1, 4)
relation[(1,4)] = "dad"
g.add_edge(2, 4)
relation[(2,4)] = "mom"

g.add_edge(1, 5)
relation[(1,5)] = "dad"
g.add_edge(2, 5)
relation[(2,5)] = "mom"

g.add_edge(3, 6)
relation[(1,6)] = "dad"
g.add_edge(4, 6)
relation[(2,6)] = "mom"

g.add_edge(3, 7)
relation[(1,7)] = "dad"
g.add_edge(4, 7)
relation[(2,7)] = "mom"

view_graph(g, relation)

My illustration window e.g. looks like that:

I have two problems and don't really know how to solve that:

Is their a possibility to show the names from the dictionary index_to_name in the illustration instead of the indexes I introduce to ensure unique nodes (in the dictionary the name Paul occurs two times and so the name "Paul" should have two nodes).
How can I fix the problem with the edge labels marked in the illustration with red rectangles. My aim is to add these labels so its clear which edge is being referenced?

I hope you can help me with this two problems.

W.r.t. problem 2), you need to correct the keys in relation. For example, g.add_edge(3, 6) is followed by relation[(1,6)] = "dad" when the presumably intended key should be relation[(3,6)] = "dad". — Paul Brodersen
– Paul Brodersen, Commented Apr 19, 2022 at 14:39

Paul Brodersen · Accepted Answer · 2022-04-19 14:49:49Z

W.r.t. problem 1), you would set with_labels to False (or simply omit the argument), and then add a function call to nx.draw_networkx_labels:

import matplotlib.pyplot as plt
import networkx as nx

def view_graph(graph, node_labels, edge_labels):
    '''
        Plots the graph
    '''
    pos = nx.spring_layout(graph)
    nx.draw(graph, pos, node_color='lightgray')
    nx.draw_networkx_labels(graph, pos, labels=node_labels)
    nx.draw_networkx_edge_labels(graph, pos, edge_labels=edge_labels)
    plt.axis('off')
    plt.show()

index_to_name = {1: "Paul", 2: "Magda", 3: "Paul", 4: "Anna", 5: "Marie", 6: "John", 7: "Mark"}

relation = {}
relation[(1, 4)] = "dad"
relation[(2, 4)] = "mom"
relation[(1, 5)] = "dad"
relation[(2, 5)] = "mom"
relation[(3, 6)] = "dad"
relation[(4, 6)] = "mom"
relation[(3, 7)] = "dad"
relation[(4, 7)] = "mom"

g = nx.from_edgelist(relation, nx.DiGraph())

view_graph(g, index_to_name, relation)

Collectives™ on Stack Overflow

Python: networkx customize graph node labels?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related