在NetworkX/Python中创建输入箭头及源汇点标识的可行性询问

当然可以实现！NetworkX本身没有内置的“特殊标识”渲染功能，但结合它常用的Matplotlib绘图工具，我们完全可以给源点、汇点设置差异化的样式——比如不同颜色、大小、形状，甚至额外的文本标注，来达到你想要的可视化效果。

第一步：识别源点与汇点 #

首先得从图中筛选出源点（针对有向图，入度为0的节点）和汇点（出度为0的节点）。这里用一个有向图示例来演示：

import networkx as nx
import matplotlib.pyplot as plt

# 创建示例有向图
G = nx.DiGraph()
G.add_edges_from([(1,2), (1,3), (2,4), (3,4), (4,5), (5,6)])

# 筛选源点（入度为0）
sources = [node for node in G.nodes if G.in_degree(node) == 0]
# 筛选汇点（出度为0）
sinks = [node for node in G.nodes if G.out_degree(node) == 0]

第二步：可视化时添加特殊标识 #

接下来我们可以给源点、汇点设置和普通节点不同的样式，比如用绿色五角星表示源点，红色圆形表示汇点，普通节点用浅蓝色：

# 固定布局（seed参数保证每次绘图位置一致）
pos = nx.spring_layout(G, seed=42)

# 绘制普通节点
nx.draw_networkx_nodes(G, pos, 
                       nodelist=[n for n in G.nodes if n not in sources + sinks],
                       node_color='lightblue', 
                       node_size=500)

# 绘制源点：绿色五角星，更大尺寸
nx.draw_networkx_nodes(G, pos, 
                       nodelist=sources,
                       node_color='green', 
                       node_size=800, 
                       node_shape='p')  # 'p'代表五角星形状

# 绘制汇点：红色圆形，更大尺寸
nx.draw_networkx_nodes(G, pos, 
                       nodelist=sinks,
                       node_color='red', 
                       node_size=800, 
                       node_shape='o')  # 'o'代表圆形

# 绘制边和节点标签
nx.draw_networkx_edges(G, pos, arrowstyle='->', arrowsize=20)
nx.draw_networkx_labels(G, pos, font_size=12, font_weight='bold')

# 调整绘图样式
plt.title("Graph with Sources (Green Pentagons) and Sinks (Red Circles)")
plt.axis('off')
plt.show()

额外扩展：添加文本标注 #

如果需要更明确的标识，还可以在源点/汇点旁边添加文字说明：

# 给每个源点添加"Source"标注
for node in sources:
    x, y = pos[node]
    plt.text(x + 0.1, y + 0.1, 'Source', fontsize=10, color='green')

# 给每个汇点添加"Sink"标注
for node in sinks:
    x, y = pos[node]
    plt.text(x + 0.1, y + 0.1, 'Sink', fontsize=10, color='red')

# 重新展示绘图
plt.title("Graph with Labeled Sources and Sinks")
plt.axis('off')
plt.show()

注意事项 #

• 如果是无向图，源点/汇点的定义可能需要根据你的业务场景调整（比如度数为1的节点），但可视化的方法完全通用——只要先确定需要特殊标识的节点列表，再用不同的绘图参数区分即可。
• NetworkX支持多种节点形状，比如s（方形）、h（六边形）等，你可以根据需求替换node_shape参数。

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