使用 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')