使用networkx的节点标签

时间:2013-03-21 13:09:08

标签: 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]

收到的图表是:

graph

代码是:

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')

1 个答案:

答案 0 :(得分:20)

您可以添加with_labels=False关键字以禁止使用networkx.draw()绘制标签,例如

networkx.draw(G, pos=pos, node_color=colors[curve],
    node_size=80, with_labels=False)

然后用

绘制特定标签
networkx.draw_networkx_labels(G,pos, labels)

其中labels是将节点ID映射到标签的字典。

看一下这个例子http://networkx.github.com/documentation/latest/examples/drawing/labels_and_colors.html