Networkx节点属性显示及get_predecessors等函数的节点数据获取问题

嘿，我来帮你搞定这两个需求！下面分两部分给你具体的实现方案，我默认你用的是最常用的NetworkX图库，如果是其他库，思路也是类似的~

一、显示节点属性（尤其是节点名称） #

要在绘图时显示节点的名称属性，核心是给绘图函数传递自定义的标签映射。举个具体的例子：

import networkx as nx
import matplotlib.pyplot as plt

# 先构建一个带name属性的示例图
G = nx.Graph()
G.add_node(1, name="用户中心")
G.add_node(2, name="订单系统")
G.add_node(3, name="支付模块")
G.add_edges_from([(1,2), (2,3)])

# 生成节点ID到name属性的映射字典
node_name_labels = {node_id: G.nodes[node_id]['name'] for node_id in G.nodes()}

# 绘图时传入labels参数，就能显示节点名称啦
nx.draw(G, labels=node_name_labels, with_labels=True, node_size=1500, font_size=12)
plt.show()

如果你的节点ID本身就是节点名称，那更简单，直接用{node: node for node in G.nodes()}作为labels参数就行。

二、通过get_predecessors/get_successors获取节点数据 #

不管是前驱还是后继节点，这些函数返回的都是节点的ID，你只需要用节点ID去图对象的nodes属性里取对应的数据就好。比如在有向图里：

# 构建一个有向图示例
DG = nx.DiGraph()
DG.add_node("A", name="登录接口", latency=0.2)
DG.add_node("B", name="权限校验", latency=0.1)
DG.add_node("C", name="数据返回", latency=0.3)
DG.add_edges_from([("A","B"), ("B","C")])

# 获取节点B的前驱节点并提取数据
predecessors = list(DG.predecessors("B"))
for pred_node in predecessors:
    # 获取该节点的所有属性
    node_data = DG.nodes[pred_node]
    print(f"前驱节点[{pred_node}]的完整数据：{node_data}")
    # 单独提取name属性
    print(f"节点名称：{node_data['name']}")

# 获取节点B的后继节点同理
successors = list(DG.successors("B"))
for succ_node in successors:
    print(f"后继节点[{succ_node}]的名称：{DG.nodes[succ_node]['name']}")

运行这段代码，你就能拿到前驱/后继节点的所有属性数据，包括名称啦。

如果是其他图库（比如PyGraphviz），逻辑也是一样的：先拿到节点对象或ID，再通过节点的属性方法获取对应的数据。

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