用值在 Python 中绘制图形

穆罕默德

我试图在图表上的箭头上显示诸如“更宽”、“更窄”之类的值，但我不知道该怎么做。我已经阅读了以前的帖子，例如如何在 python 中使用 networkx 绘制有向图？并阅读NetworkX的文档（https://networkx.org/documentation/latest/_downloads/networkx_reference.pdf），但我无法实现。

下面提到了我的代码；

# 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

您可以使用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()