从python集类型构造图

时间:2010-03-19 05:36:30

标签: python graph set nodes edges

简短的问题,是否有一个自我功能从一组python集合制作图表? 更长的问题:我有几个python集。它们各自重叠,或者一些是其他的子集。我想制作一个图形(如在节点和边缘中)节点是集合中的元素。边是集合的交集,并且通过集合的交集中的元素数量加权。 python有几个图形包。 (NetworkX,igraph,...)我不熟悉其中任何一个的使用。他们是否会直接从集合列表中创建图表,即MakeGraphfromSets(alistofsets) 如果不是,您知道如何获取集合列表来定义边缘的示例。它实际上看起来可能是直截了当的但是一个例子总是很好。

2 个答案:

答案 0 :(得分:2)

自己编码并不难:

def intersection_graph(sets):
    adjacency_list = {}
    for i, s1 in enumerate(sets):
        for j, s2 in enumerate(sets):
            if j == i:
                continue
            try:
                lst = adjacency_list[i]
            except KeyError:
                adjacency_list[i] = lst = []
            weight = len(s1.intersection(s2))
            lst.append( (j, weight) )
    return adjacency_list

此函数为每个集合编号,其索引在sets内。我们这样做是因为dict键必须是不可变的,这对于整数而不是集合都是正确的。

以下是如何使用此功能的示例,以及它的输出:

>>> sets = [set([1,2,3]), set([2,3,4]), set([4,2])]
>>> intersection_graph(sets)
{0: [(1, 2), (2, 1)], 1: [(0, 2), (2, 2)], 2: [(0, 1), (1, 2)]}

答案 1 :(得分:2)

def MakeGraphfromSets(sets):
    egs = []
    l = len(sets)
    for i in range(l):
        for j in range(i,l):
            w = sets[i].intersection(sets[j])
            egs.append((i,j,len(w)))
    return egs

# (source set index,destination set index,length of intersection)

sets = [set([1,2,3]), set([2,3,4]), set([4,2])]

edges = MakeGraphfromSets(sets)

for e in edges:
    print e

输出:

(0, 0, 3)
(0, 1, 2)
(0, 2, 1)
(1, 1, 3)
(1, 2, 2)
(2, 2, 2)