使用 NetworkX 和 Python 对图的边进行分区

Question

我想使用 Python 和 NetworkX 根据边缘属性对图 g 的边缘进行分区。在此代码片段中：

import networkx as nx

g = nx.Graph()
g.add_node(1, pos=[0, 0])
g.add_node(2, pos=[0, 2])
g.add_node(3, pos=[1, 1])
g.add_node(4, pos=[2, 1])
g.add_edge(1, 3, cat='a')
g.add_edge(4, 3, cat='a')
g.add_edge(1, 2, cat='b')
g.add_edge(4, 2, cat='b')

我想根据 cat 属性将图 g 划分为图 ga 和 gb 并保留节点属性pos。请注意，节点 1 和 4 将同时属于 ga 和 gb。有库支持此操作吗？

Answer 1

您可以使用列表理解和集合的组合：

ga = g.subgraph(set(*[[u,v] for (u,v,data) in g.edges(data=True) if data.get('cat')=='a']))
gb = g.subgraph(set(*[[u,v] for (u,v,data) in g.edges(data=True) if data.get('cat')=='b']))

Answer 2

这似乎是一个简单的解决方案：

ga = g.edge_subgraph(e for e in g.edges() if g.edges[e]['cat']=='a')
gb = g.edge_subgraph(e for e in g.edges() if g.edges[e]['cat']=='b')