在igraph [python]中将两个有向边折叠成一个

时间:2016-12-20 23:38:24

标签: python igraph

我有一个有向图,其中在许多情况下,在两个节点之间存在边缘双向,具有不同的权重。例如,a - > b的权重为3,b - > a的权重为1.我想将这些组合成一个只存在聚合权重的图表,即这个例子会产生一个图表,其中a- - >体重2.

我见过g.simplify(),但是如何让它结合相反的边缘呢?

1 个答案:

答案 0 :(得分:3)

假设在相同的两个顶点之间可以有相同方向的多个边,并且如果是这样,你想要加权,你可以从

开始
g.simplify(combine_edges='sum')

然后对于每对顶点 a b ,最多一条边从 a b

要将这些相对边组合成一条边,您可以使用实例方法to_undirected,它将g转换为无向图。在第一步之后,每对顶点之间应该至多有两条边,总是相对的。因此,如果你想从另一个方向减去一个方向的重量,你可以这样做:

def subwt(attrs):
     if len(attrs) == 1:
         return attrs[0]
     assert len(attrs) == 2
     return attrs[0] - attrs[1]

g.to_undirected(combine_edges=subwt)

但是这会给你一个无向图,以及边的权重是否应该从 a b b to a ,是不可能的。

如果你想要一个有向图,你如何选择是否从权重为2的 a b 或从 b a ,重量为-2?

这是一个能产生这种有向图的函数;输出图中每条边的方向由输入图中顶点之间遇到的第一条边确定。此外,任何边缘属性除了' weight'从第一条边复制,忽略任何其他边。

def combine_edges(graph):
    combe = graph.copy() # copies graph attributes, vertex attributes
    combe.delete_edges(None) # removes all edges
    for e in graph.es:
        src, targ = e.tuple
        if combe.are_connected(src, targ):
            ced = combe.es(_source=src, _target=targ)[0]
            ced['weight'] += e['weight']
        elif combe.are_connected(targ, src):
            ced = combe.es(_source=targ, _target=src)[0]
            ced['weight'] -= e['weight']
        else:
            combe.add_edge(src, targ, **e.attributes())
    return combe