Question

我正在为整数分区编写代码并构建一个图，其中每个节点都是一个分区。我想用图像中的节点标记{2,1,1}，{1,1,1,1}，{2,2}等分区元素。

所以我想知道如何在networkx中标记节点。

我已经看过标记节点的代码，但我没有得到它。代码如下：

nx.draw_networkx_nodes(G,pos,
                       nodelist=[0,1,2,3],
                       node_color='r',
                       node_size=500,
                    alpha=0.8)

但我想要的是标记已经构建的图的各个节点！

我在这里提供我的代码，以便您可以更好地理解它。

import networkx as nx
from matplotlib import pylab as pl

def partitions(num):
    final=[[num]]

    for i in range(1,num):
        a=num-i
        res=partitions(i)
        for ele in res:
            if ele[0]<=a:
                final.append([a]+ele)

    return final

def drawgraph(parlist):
    #This is to draw the graph
    G=nx.Graph()
    length=len(parlist)
    print "length is %s\n" % length

    node_list=[]

    for i in range(1,length+1):
        node_list.append(i)

    G.add_cycle(node_list)

    nx.draw_circular(G)
    pl.show()

请帮帮我。

非常感谢

Answer 1

由于node_list由整数组成，因此您的节点将这些整数的字符串表示形式作为标签。但是您的节点可以是任何可清除对象，而不仅仅是整数。因此，最简单的方法是使node_list成为parlist中项目的字符串表示形式。（parlist中的项目是列表，这些列表是可变的，因此不可清除。这就是我们不能仅使用parlist作为node_list的原因。）

还有函数nx.relabel_nodes，我们可以使用它，但我认为首先给节点提供正确的标签更简单。

import networkx as nx
import matplotlib.pyplot as plt


def partitions(num):
    final = [[num]]

    for i in range(1, num):
        a = num - i
        res = partitions(i)
        for ele in res:
            if ele[0] <= a:
                final.append([a] + ele)

    return final


def drawgraph(parlist):
    G = nx.Graph()
    length = len(parlist)
    print "length is %s\n" % length
    node_list = [str(item) for item in parlist]
    G.add_cycle(node_list)
    pos = nx.circular_layout(G)
    draw_lifted(G, pos)


def draw_lifted(G, pos=None, offset=0.07, fontsize=16):
    """Draw with lifted labels
    http://networkx.lanl.gov/examples/advanced/heavy_metal_umlaut.html
    """
    pos = nx.spring_layout(G) if pos is None else pos
    nx.draw(G, pos, font_size=fontsize, with_labels=False)
    for p in pos:  # raise text positions
        pos[p][1] += offset
    nx.draw_networkx_labels(G, pos)
    plt.show()

drawgraph(partitions(4))

产量

enter image description here

如何在networkx中的图形中为节点添加标签？

1 个答案: