python - 在 python 中使用 networkx 创建树数据

Question

我正在尝试创建一棵具有 1111 个节点的树，其组织方式如下节点 1 有 10 个子节点(2 到 11)，节点 2 有 10 个子节点(12 到 21)等等...这样每个节点有 10 个子节点，在根级别有 1 个节点，有 10 个子节点，每个子节点有 10 个 child ，这 100 个节点中的每一个都有 10 个子节点，每个子节点有 1000 个叶节点。节点总数为1111。

import networkx as nx
G = nx.Graph()
L1 = [1]
L2 = [x for x in range(2,12)]
L3 = [x for x in range(12,112)]
L4 = [x for x in range(112,1112)]

G.add_node(1)
G.add_nodes_from(L1)
G.add_nodes_from(L2)
G.add_nodes_from(L3)
G.add_nodes_from(L4)

现在我想使用 G.add_edges_from([(x,y) for x in L1 for y in L2]) 添加边，这对第一层没问题，但我该怎么做对于其他级别？

Answer 1

您可以在一行中“开箱即用”地获得所需的结果:

import networkx as nx

G = nx.balanced_tree(10,10)