Plot Network Graph and add edge weights to network edges python

Question

I am trying to plot the network Graph with a sample data as below:-

    From    To  Density
0   A       B    296
1   B       C    258
2   C       D    296
3   D       E    274
4   E       F    272
5   F       G    195
6   G       H    286
7   H       I    257
8   I       J    204
9   J       K    66

I want to add the Density number on the edges like how many times A to B has been done and the same goes to rest. Each edge should be having the Density number above the node edge/arrow.

This is what I have tried so far:-

G = nx.from_pandas_edgelist(network_data,'FROM','TO', edge_attr='COUNT',create_using=nx.DiGraph())

fig, ax = plt.subplots(figsize=(20,15))
pos = nx.kamada_kawai_layout(G)
nx.draw_networkx_nodes(G,pos, ax = ax,node_size=1500)
nx.draw_networkx_edges(G, pos, ax=ax)
labels = nx.get_edge_attributes(G,'COUNT')
_ = nx.draw_networkx_labels(G, pos, ax=ax)
nx.draw_networkx_edge_labels(G,pos,edge_labels=labels)
plt.show()

But this solution won't give the arrows it just give me the weight between 2 nodes. and also as the data is heavy the plot looks clumsy.

I have seen some of the solutions where the width of the edges is increased but that didn't work in my case as i have 3 digit numbers for Density. I am really new to this and some help will be appreciated. Thanks in advance

@ Let's try I have edited the question with what i have tried. please take a look — nivedan gowda
– nivedan gowda, Commented Aug 27, 2020 at 12:24

OmriKaduri · Accepted Answer · 2020-08-27 12:26:08Z

3

You can use networkx.

Specifically, you can create a networkx graph from a dictionary, as:

import networkx as nx

dod = {'A': {'B': {"weight": 256}},
       'B':{'C':{'weight':258}},
       'C':{'D':{'weight':296}}}

G = nx.from_dict_of_dicts(dod)

If you'll want to draw:

pos=nx.spring_layout(G)
labels = nx.get_edge_attributes(G,'weight')
nx.draw(G,pos)
nx.draw_networkx_edge_labels(G,pos,edge_labels=labels)

edited Aug 27, 2020 at 12:26

answered Aug 27, 2020 at 11:29

OmriKaduri

3161 silver badge8 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

nivedan gowda Over a year ago

Thanks a lot for your answer. But as I already mentioned in the question i don't want the thickness of the edges to be increased i want the exact number on top of the edge arrow.

OmriKaduri Over a year ago

I've edited my answer for that drawing (A minor change from draw_networkx_edges to draw_networkx_edge_labels)

Collectives™ on Stack Overflow

Plot Network Graph and add edge weights to network edges python

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related