Question

我不知道正确的术语，但我在networkx中有一个有向图，我想使用映射函数合并相关节点，保持到原始节点的链接

例如，如果我有这个图表：

a1 -> b1
a2 -> b2
a1 -> c1
a2 -> d2
b1 -> d2
c1 -> f1
c2 -> f2
b1 -> c2

和这个功能

def mymap(node):
    return node[0]

然后我想得到这个图：

a -> b
a -> c
a -> d
b -> d
c -> f
b -> c

使用节点数据

a: original_nodes = [a1, a2]
b: original_nodes = [b1, b2]
c: original_nodes = [c1, c2]
d: original_nodes = [d2]
f: original_nodes = [f1, f2]

有办法做到这一点吗？（没有自己手动迭代节点和边缘;我可以这样做，但我认为这将是一项常见的任务）

Answer 1

嗯，我以为我以前记得这样做过，只是在我的一个Python脚本中找到了这个：

def supergraph(g1, keyfunc, allow_selfloops=True):
    g2 = nx.DiGraph()
    for (a,b,d) in g1.edges_iter(data=True):
        result = keyfunc(g1,a,b,d)
        if result is not None:
            a2,b2,w = result
            if a2 != b2 or allow_selfloops:
                g2.add_edge(a2,b2)
                try:
                    g2[a2][b2]['weight'] += w
                except:
                    g2[a2][b2]['weight'] = w

            for u2,u in [(a2,a),(b2,b)]:
                if not u2 in g2:
                    g2.add_node(u2, original_nodes=set([u]))
                else:
                    try:
                        g2.node[u2]['original_nodes'].add(u)
                    except:
                        g2.node[u2]['original_nodes'] = set([u])
    return g2

其中keyfunc(g,a,b,d)与图g，节点a和b以及边数据d是一个映射函数，应该返回None（忽略该边）或元组{{1使用新节点a2和b2，以及权重w。

在networkx中对节点进行分组

1 个答案: