3

我正在为整数分区编写代码并构建一个图，其中每个节点都是一个分区。我想用 {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()

请帮我。

非常感谢

4

1 回答 1

7

由于您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))

产量

在此处输入图像描述

于 2013-07-18T10:05:43.303 回答