在图上显示两个节点之间的边数

Question

我正在从数据帧制作网络图，如下所示import networkx as nximport pandas as pdimport matplotlib.pyplot as pltdata = {'source': ['A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A'],        'target': ['B', 'B', 'C', 'D', 'E', 'B', 'C', 'E', 'F', 'C']}df = pd.DataFrame(data=data)G = nx.from_pandas_edgelist(df, source='source', create_using=nx.Graph(), target='target')pos = nx.spring_layout(G)nx.draw_networkx(G, pos, with_labels=False, node_color='#1f0ee3', node_size=150,                 edge_color='r', width=1, alpha=0.6, style='dashed',                 font_color='k', font_size=15, font_weight=20)# Offset for node labelsoffset = 0.15pos_labels = {}keys = pos.keys()for key in keys:    x, y = pos[key]    pos_labels[key] = (x, y + offset)nx.draw_networkx_labels(G, pos=pos_labels, fontsize=2)plt.xlim(-1.25, 1.25)plt.ylim(-1.25, 1.25)plt.title('Example')plt.show()它产生然而，这并没有显示的是每个节点之间的连接情况。例如，A->B实际上有三条边。 会使一些边缘看起来比其他边缘更厚，但我真正想要的是图上显示的两个节点之间的实际边缘数。这样，“B”的标签是“B（3）”（或类似的东西）。nx.MultiGraph()我知道这可以告诉我节点之间的边数，但我想以编程方式将两个节点之间的边数显示为绘图上的计数。nx.Graph.number_of_edges()

Answer 1

您可以调整数据帧以在目标节点字符串中包含边缘编号：

data = {'source': ['A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A', 'A'],

        'target': ['B', 'B', 'C', 'D', 'E', 'B', 'C', 'E', 'F', 'C']}

df = pd.DataFrame(data=data)

df = df.groupby(df.columns.tolist()).size().reset_index().rename(columns={0:'edges'})

df['target'] = df['target'] + ' (' + df['edges'].astype(str) + ')'

Answer 2

可用于标记边。提供一个字典，其中边缘作为键，边缘数作为值：nx.draw_networkx_edge_labels

counts = df.groupby('source')['target'].value_counts().to_dict()

nx.draw_networkx_edge_labels(G, pos, edge_labels=counts)