使用networkx的节点标签
问题描述
我正在根据curveSeq
保存的给定的Y值序列创建一个图。(自动枚举X值:0,1,2...)
即对于curveSeq = [10,20,30]
,我的图形将包含以下点:
<0,10>, <1,20>, <2,30>.
我在同一张nx.Graph
上绘制了一系列图表,以便在一张图片中显示所有内容。
我的问题是:
- 每个节点都表示其位置。即
<0,10>
位置中的节点显示其各自的标签,我不知道如何删除它。 - 存在我要向其添加标签的特定节点,但我不知道如何添加。
例如,对于序列:
[0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9,1,1]
收到的图形为:
代码为:
for point in curveSeq:
cur_point = point
#assert len(cur_point) == 2
if prev_point is not None:
# Calculate the distance between the nodes with the Pythagorean
# theorem
b = cur_point[1] - prev_point[1]
c = cur_point[0] - prev_point[0]
a = math.sqrt(b ** 2 + c ** 2)
G.add_edge(cur_point, prev_point, weight=a)
G.add_node(cur_point)
pos[cur_point] = cur_point
prev_point = cur_point
#key:
G.add_node((curve+1,-1))
pos[(curve+1,-1)] = (curve+1,-1)
nx.draw(G, pos=pos, node_color = colors[curve],node_size=80)
nx.draw_networkx_edges(G,pos=pos,alpha=0.5,width=8,edge_color=colors[curve])
plt.savefig(currIteration+'.png')
解决方案
您可以使用networkx.draw()
添加with_labels=False
关键字来抑制标签绘制,例如
networkx.draw(G, pos=pos, node_color=colors[curve],
node_size=80, with_labels=False)
然后用
绘制特定标签networkx.draw_networkx_labels(G,pos, labels)
其中Labels是将节点ID映射到标签的词典。
查看此示例:https://networkx.org/documentation/stable/auto_examples/drawing/plot_labels_and_colors.html
相关文章