Plotting a graph in Python with Values

Question

I am trying to show the values such as "Broader","Narrower" on the arrows on a graph but i dont know how to do it. I have read the previous posts such as how to draw directed graphs using networkx in python? and read the documentation (https://networkx.org/documentation/latest/_downloads/networkx_reference.pdf) of NetworkX but i cant make it happen.

My code is mentioned below;

# libraries
import pandas as pd
import numpy as np
import networkx as nx
import matplotlib.pyplot as plt
 
# Build a dataframe with your connections
df = pd.DataFrame({ 'from':['Plankton', 'Plankton', 'Plankton', 'Plankton', 'Plankton', 'Plankton', 'Plankton'], 'to':['aquatic communities', 'plankton surveys', 'zooplankton','phytoplankton', 'cryoplankton', 'nannoplankton', 'picoplankton'], 'value':['broader', 'related', 'narrower', 'narrower','narrower','narrower','narrower']}) 
# And I need to transform my categorical column in a numerical value typeA->1, typeB->2...
df['value']=pd.Categorical(df['value'])
df['value'].cat.codes
 
# Build your graph
G=nx.from_pandas_edgelist(df, 'from', 'to', create_using=nx.DiGraph(directed=True) )
 
# Custom the nodes:

nx.draw_networkx(G, font_size = 10, with_labels=True, arrows=True, node_color= 'skyblue', node_size= 500, width= 3.5, arrowstyle= '-|>', arrowsize= 12, edge_color=df['value'].cat.codes)

Ynjxsjmh · Accepted Answer · 2021-04-18 05:12:53Z

0

You can draw edge labels with draw_networkx_edge_labels().

G = nx.from_pandas_edgelist(df, 'from', 'to', create_using=nx.DiGraph(directed=True) )

pos = nx.spring_layout(G)

nx.draw_networkx(G, pos, font_size = 10, with_labels=True, arrows=True, node_color= 'skyblue', node_size= 500, width= 3.5, arrowstyle= '-|>', arrowsize= 12, edge_color=df['value'].cat.codes)

nx.draw_networkx_edge_labels(G,pos,edge_labels=dict(zip(G.edges, df['value'].tolist())))

plt.show()

It has extra optional argument label_pos for you to adjust the label position in edge.

0: head
0.5: center
1: tail

Other float between 0, 1 is also supported.

answered Apr 18, 2021 at 5:12

Ynjxsjmh

30.3k7 gold badges43 silver badges64 bronze badges

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

Collectives™ on Stack Overflow

Plotting a graph in Python with Values

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related