Python 是一門高級編程語言,其靈活和易用性使得處理各種數據類型和編寫各種算法變得非常容易。它同時也擁有豐富的第三方庫,使得 Python 可以在各個領域都有著廣泛的應用。本文將介紹如何使用 Python 來畫樹,并添加注釋。
# 導入必要的庫 import matplotlib.pyplot as plt import networkx as nx # 創建一顆樹 tree = nx.balanced_tree(2, 4) # 可以通過這種方式來指定樹型結構:tree = nx.DiGraph([(0,1),(0,2),(1,3),(1,4),(2,5),(2,6)]) # 添加一個布局 pos = nx.layout.spring_layout(tree) # 創建一個 Figure fig = plt.figure() # 創建一個 Axes,并繪制樹 ax = fig.add_subplot(111) nx.draw(tree, pos=pos, ax=ax) # 添加節點的注釋 for node in tree: ax.annotate(str(node), xy=pos[node], ha='center', va='center') # 顯示圖像 plt.show()
上述代碼中的第 3 行創建了一顆平衡樹,它包含了 15 個節點,其中樹的深度為 4。第 6 行指定了布局,本例中我們采用的是 Spring 布局;第 8 行創建了一個 Figure 對象;第 10 行創建了一個 Axes 并繪制了樹;第 13~16 行用于添加節點的注釋,其中 annotate 方法為注釋添加了節點的名稱,并設置了注釋的位置以及文字的對齊方式;第 19 行用于顯示圖像。
當運行上述代碼后,我們可以得到一個如下圖所示的帶有注釋的樹狀結構。
![帶有注釋的樹](https://i.imgur.com/FjK5m5G.png)