Python 如何以id作为节点标签绘制树

如果使用networkx生成树，默认情况下，节点具有整数ID。我想用标记节点的整数来可视化树。因此，不是：

import matplotlib.pyplot as plt
import networkx as nx
T = nx.generators.balanced_tree(2, 2)
nx.draw(T)
plt.show()

---

如何在节点内写入整数标签？

使用

nx.draw

，而不是

nx.draw\u networkx

。它有一个带有标签的关键字参数

，默认情况下为True，因此除了函数之外，您不需要更改任何内容

import matplotlib.pyplot as plt
import networkx as nx
T = nx.generators.balanced_tree(2, 2)
nx.draw_networkx(T)
plt.show()

或只是
nx.draw（带标签=True）
可以说更干净，因为它不会改变其他默认值。