从 python 集合类型构造图

Question

简单的问题是，是否有一个独立的函数可以从 python 集合的集合中制作图表？ 更长的问题：我有几套Python。它们各自重叠或者有些是其他的子集。我想制作一个图表（如节点和边），节点是集合中的元素。边是集合的交集，并按集合交集中的元素数量加权。 python 有几个绘图包。 （NetworkX，igraph，...）我不熟悉其中任何一个的使用。他们中的任何一个都会直接从集合列表中制作图形，即 MakeGraphfromSets(alistofsets) 如果没有，您是否知道如何使用集合列表来定义边缘的示例。实际上看起来可能很简单，但有一个例子总是好的。

Answer 1

自己编写代码并不难：

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 中的索引对每个集合进行编号。我们这样做是因为字典键必须是不可变的，这对于整数来说是正确的，但对于集合则不然。

这是如何使用此函数的示例及其输出：

>>> 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)]}

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