使用networkx的节点标签

2022-03-31 00:00:00 python matplotlib 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

相关文章