如何用Python的NetworkX实现指定参数的多层节点图？

需求说明 #

• 起始节点为0，参数包含n（节点度）和l（层数）
• 第1层：节点0连接n个全新节点，构成第1层
• 中间层：每层的每个节点需要连接n-1个全新节点，保证自身节点度为n
• 最后一层：所有节点仅与父节点相连，节点度为1

原尝试的代码无法适配更大的n和l值，例如当n=3、l=2时能达到预期，但新增层数后无法正确扩展。

原代码问题 #

原代码的循环逻辑错误，没有按分层方式管理节点，节点计数和连接规则完全不符合需求，导致无法正确生成符合规则的图。

通用实现思路 #

1. 分层管理节点：用列表存储每一层的节点集合，方便遍历和处理
2. 节点计数跟踪：维护全局节点ID计数器，确保每次新增节点都是全新的
3. 按层构建连接：
   • 第0层（起始层）仅包含节点0
   • 第1层：从节点0出发，创建n个新节点并建立连接
   • 第2到第l层：遍历上一层的每个节点，为其创建n-1个新节点并建立连接（父节点已有一个向上的连接，需n-1个新节点满足度为n的要求）
   • 最后一层节点无需创建子节点，自然度为1

完整实现代码 #

import networkx as nx
import matplotlib.pyplot as plt

def build_hierarchical_graph(n, l):
    G = nx.Graph()
    # 初始化：第0层只有节点0
    layers = [[0]]
    current_node_id = 1  # 下一个要创建的节点ID
    
    # 构建第1层到第l层
    for layer_idx in range(1, l+1):
        current_layer = []
        # 获取上一层的所有节点
        prev_layer = layers[layer_idx-1]
        
        for node in prev_layer:
            # 确定当前节点需要创建的子节点数量：第1层父节点是0，需n个；其他层需n-1个
            num_children = n if layer_idx == 1 else (n - 1)
            # 创建子节点并建立连接
            for _ in range(num_children):
                G.add_edge(node, current_node_id)
                current_layer.append(current_node_id)
                current_node_id += 1
        layers.append(current_layer)
    
    return G

# 示例：n=3，l=3
n = 3
l = 3
G = build_hierarchical_graph(n, l)

# 绘图展示
plt.figure(figsize=(10, 10))
nx.draw_networkx(G, with_labels=True, node_size=700, font_size=12)
plt.show()

代码解释 #

• layers列表：按顺序存储每一层的节点，方便后续遍历处理
• current_node_id：确保每次新增节点唯一，避免重复
• 分层循环：针对第1层和后续层分别处理子节点数量，严格遵循节点度规则
• 示例中n=3、l=3时，第1层有3个节点（1、2、3），第2层每个节点创建2个新节点（共6个节点：4-9），第3层每个节点再创建2个新节点（共12个节点：10-21），完全符合需求

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