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

Question

我正在从数据框制作网络图，如下所示

import networkx as nx
import pandas as pd
import matplotlib.pyplot as plt

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)

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 labels
offset = 0.15
pos_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 实际上有三个边。nx.MultiGraph()会使某些边缘看起来比其他边缘更厚，但我真正想要的是绘图上显示的两个节点之间的实际边缘数。这样'B'的标签是'B（3）'（或类似的东西）。

我知道这nx.Graph.number_of_edges()可以告诉我节点之间的边数，但我想以编程方式显示两个节点之间的边数作为绘图的计数。

标签： pythonpandasnetworkx

Answer 1

您可以使用nx.draw_networkx_edge_labels标签边缘。提供一个以边为键、边数为值的字典：

counts = df.groupby('source')['target'].value_counts().to_dict()
nx.draw_networkx_edge_labels(G, pos, edge_labels=counts)