Networkx中如何筛选含指定属性的节点（避免KeyError）

当部分节点未定义目标属性时，直接通过字典键访问会触发KeyError，可以通过以下方式实现安全的属性筛选：

基础版：筛选拥有指定属性的节点列表 #

定义函数检查每个节点的属性字典是否包含目标属性，避免键不存在的错误：

import networkx as nx

def get_nodes_with_attribute(graph, attr_name):
    return [node for node, attrs in graph.nodes(data=True) if attr_name in attrs]

测试示例 #

P = nx.Graph()
P.add_node("node1", at=5)
P.add_node("node2")
P.add_node("node3")

# 获取所有含'at'属性的节点
target_nodes = get_nodes_with_attribute(P, "at")
print(f"拥有'at'属性的节点：{target_nodes}")
# 输出：拥有'at'属性的节点：['node1']

进阶版：筛选拥有指定属性且属性值匹配的节点 #

如果需要同时匹配属性值，可扩展函数：

def get_nodes_with_attribute_value(graph, attr_name, attr_value):
    return [
        node 
        for node, attrs in graph.nodes(data=True) 
        if attr_name in attrs and attrs[attr_name] == attr_value
    ]

测试示例 #

# 获取含'at'属性且值为5的节点
target_nodes = get_nodes_with_attribute_value(P, "at", 5)
print(f"拥有'at'属性且值为5的节点：{target_nodes}")
# 输出：拥有'at'属性且值为5的节点：['node1']

核心逻辑说明 #

• 使用graph.nodes(data=True)遍历节点时，会返回(节点名, 属性字典)的元组
• 通过attr_name in attrs先判断属性是否存在，避免直接访问不存在的键引发KeyError

内容的提问来源于stack exchange，提问作者Robert Alexander