Làm thế nào để điều chỉnh độ dài các nhánh của một biểu đồ dendrogram trong Matplotlib?

Để điều chỉnh độ dài nhánh của hình ảnh dendrogram trong Matplotlib, chúng ta có thể thực hiện các bước sau -

• Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô con.

• Vẽ mẫu ngẫu nhiên ( a và b ) từ phân phối chuẩn đa biến.

• Nối chuỗi các mảng dọc theo một trục hiện có, sử dụng nối () phương pháp.

• Thực hiện phân cụm phân cấp / kết tụ.

• Tạo một hình mới hoặc kích hoạt một hình hiện có bằng cách sử dụng figure () phương pháp.

• Thêm các trục vào hình như một phần của sắp xếp ô con.

• Vẽ đồ thị phân nhóm theo thứ bậc dưới dạng biểu đồ dendrogram bằng cách sử dụng dendrogram () phương pháp.

• Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

from matplotlib import pyplot as plt
from scipy.cluster.hierarchy import dendrogram, linkage
import numpy as np
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
a = np.random.multivariate_normal([0, 10], [[3, 1], [1, 4]], size=[2, ])
b = np.random.multivariate_normal([0, 10], [[3, 1], [1, 4]], size=[3, ])
X = np.concatenate((a, b), )
Z = linkage(X)
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
dendrogram(Z, ax=ax)
plt.show()

Đầu ra