networkx怎么给节点添加标签

Question

import networkx as nx
import matplotlib.pyplot as plt


# 数据准备
G = nx.DiGraph(with_labels=True)  # 单边有向图

G.add_edges_from([(1,2), ('x','y')]) # 一次性添加多条边
nx.draw(G)        # 简单画一下


plt.axis("off")           # 去掉边框
plt.show()

Answer 1

可以通过 nx.set_node_attributes() 给节点添加标签，具体实现如下：

import networkx as nx
import matplotlib.pyplot as plt
 
# 数据准备
G = nx.DiGraph(with_labels=True)  # 单边有向图
G.add_edges_from([(1,2), ('x','y')]) # 一次性添加多条边
 
# 添加节点标签
labels = {1: 'A', 2: 'B', 'x': 'C', 'y': 'D'}
nx.set_node_attributes(G, labels, 'label')
 
# 绘制图形
pos = nx.spring_layout(G)
nx.draw(G, pos)
nx.draw_networkx_labels(G, pos, labels=nx.get_node_attributes(G, 'label'))
plt.axis("off")
plt.show()

其中，nx.set_node_attributes() 的三个参数分别为图对象、标签字典和标签名称。在绘制图形时，使用 nx.draw_networkx_labels() 绘制节点标签。