溫馨提示×

如何在Matplotlib中繪制樹狀圖或層次聚類圖

小億
139
2024-05-20 13:05:32
欄目: 編程語言

在Matplotlib中繪制樹狀圖或層次聚類圖可以使用scipy庫中的dendrogram函數來實現。以下是一個簡單的示例代碼:

import numpy as np
from scipy.cluster.hierarchy import dendrogram, linkage
import matplotlib.pyplot as plt

# 生成隨機數據
data = np.random.rand(10, 2)

# 計算層次聚類
Z = linkage(data, 'ward')

# 繪制樹狀圖
plt.figure(figsize=(10, 5))
dendrogram(Z)
plt.title('Hierarchical Clustering Dendrogram')
plt.xlabel('Sample Index')
plt.ylabel('Distance')
plt.show()

在上面的代碼中,首先生成了一些隨機數據,然后使用linkage函數計算數據的層次聚類,并最后通過dendrogram函數繪制樹狀圖。

您可以根據自己的數據和需求對代碼進行修改和調整,以實現不同類型和樣式的樹狀圖或層次聚類圖。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女