在Networkx中获取特定类型的节点[重复]

Question

所以我使用 networkX 图来表示一些信息。此信息由不同的对象类型（例如，ColorNode 和 ShapeNode）表示。

在此图上完成的一些处理需要我提取出特定类型的节点。每次我需要这样做时，我都会按照下面的代码行做一些事情。

colornodes = []
for node in graph.nodes():
    if isinstance(node, ColorNode):
        colornodes.append()

虽然这可行，但我觉得这是在处理图表时经常出现的情况，我正在那里重新发明轮子。 本质上，我想知道是否有更好的方法来做到这一点。

Answer 1

我建议另一种方法，而不是定义自己的类型并始终检查isinstance（这非常慢）。

您可以查看此answer 以了解经典节点/边缘过滤。

但是我发现另一个技巧可能对您的具体情况派上用场。

如果您定义了一个代表节点类型的属性，您可以使用以下方法查询具有该特定属性的节点 内置get_node_attributes 函数。诀窍是它只返回真正定义属性的节点：

import networkx as nx

G = nx.complete_graph(10)
G.node[0]['ColorNode'] = True  # right-hand side value is irrelevant for the lookup
G.node[1]['ColorNode'] = True
G.node[2]['ShapeNode'] = True
G.node[3]['ShapeNode'] = True

shape_nodes = nx.get_node_attributes(G, 'ShapeNode').keys()
color_nodes = nx.get_node_attributes(G, 'ColorNode').keys()

print('Shape node ids: {}'.format(shape_nodes))
print('Color node ids: {}'.format(color_nodes))

输出：

Shape node ids: [2, 3]
Color node ids: [0, 1]

当然，如果您的图表很大或静态，您应该保留 id 列表以便快速查询！

Answer 2

而不是graph.nodes()， 你可以使用 xpath。