根据节点的属性NetworkX将图划分为太阳图

Question

我正在使用Networkx计算图形的某些度量，例如直径，聚类系数等。直接对整个图形执行此操作很简单。我感兴趣的是在具有相同属性（例如颜色）的节点之间找到这些度量。我在考虑是否可以将图形划分为不同的子图形，其中每个子图形中的节点具有相同的颜色，那么我可以继续进行并测量该子图形中的直径。所以我的问题是：是否有一种方法可以将图划分为包含相同颜色节点的子图？

我将不胜感激。

Answer 1

采用
Graph.subgraph(nodes)

演示版

import networkx as nx

G = nx.Graph()

G.add_nodes_from([1, 2, 3], color="red")
G.add_nodes_from([4, 5, 6])
G.nodes  # NodeView((1, 2, 3, 4, 5, 6))

# create generator
nodes = (
    node
    for node, data
    in G.nodes(data=True)
    if data.get("color") == "red"
)
subgraph = G.subgraph(nodes)
subgraph.nodes  # NodeView((1, 2, 3))

旧版NetworkX

遍历（Graph.iter_nodes()）并根据您的条件过滤节点。将该列表传递给Graph.subgraph()，它将返回这些节点及其内部边缘的副本。

例如：

G = nx.Graph()
# ... build or do whatever to the graph
nodes = (n for n, d in G.nodes_iter(data=True)) if d.get('color') == 'red')
subgraph = G.subgraph(nodes)