如何使用python的networkx模块从节点列表生成完全连接的子图

时间:2012-05-18 09:02:02

标签: python networkx

我需要使用networkx生成一个完全连接的子图,从我想要连接的节点列表开始。基本上,我希望传递给函数的列表中的所有节点都相互连接。

我想知道是否有任何内置函数来实现这一点(我还没有找到)? 或者我应该考虑一些算法?

非常感谢你。

3 个答案:

答案 0 :(得分:13)

我不知道有任何方法可以做到这一点,但你可以很容易地模仿networkx的complete_graph()方法并略微改变它(几乎像内置):

import networkx
import itertools

def complete_graph_from_list(L, create_using=None):
    G = networkx.empty_graph(len(L),create_using)
    if len(L)>1:
        if G.is_directed():
            edges = itertools.permutations(L,2)
        else:
            edges = itertools.combinations(L,2)
        G.add_edges_from(edges)
    return G

S = complete_graph_from_list(["a", "b", "c", "d"])
print S.edges()

答案 1 :(得分:10)

有一个用于创建完全连接(即完整)图形的函数,nameley complete_graph

import networkx as nx
g = nx.complete_graph(10)

它需要一个整数参数(图中的节点数),因此您无法控制节点标签。我没有找到自动执行此操作的功能,但使用itertools它很容易:

from itertools import combinations

nodes = ['A', 'B', 'C', 'D', 'E']
edges = combinations(nodes, 2)
g = nx.Graph()
g.add_nodes_from(nodes)
g.add_edges_from(edges)

combinations(nodes, 2)将创建包含nodes的所有对组合的2元素元组,然后它们将作为图形中的边缘。

此解决方案仅适用于无向图。请查看zubinmehta's solution以获得更一般的方法。

答案 2 :(得分:4)

您可以使用networkx命令直接生成包含整数节点的clique,然后有一个简单的命令可以将节点重新标记为任何其他可哈希名称。

import networkx as nx
L=["hello", "world", "how", "are", "you"]
G=nx.complete_graph(len(L))
nx.relabel_nodes(G,dict(enumerate(L)), copy = False) #if copy = True then it returns a copy.