Scipy树状图的简单着色

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

# 生成随机数据
np.random.seed(0)
X = np.random.random((10, 2))

# 使用层次聚类算法进行聚类
Z = linkage(X, method='ward')

# 生成树状图
plt.figure(figsize=(10, 5))
dendrogram(Z, color_threshold=0.5)

# 设置节点颜色
colors = ['r', 'g', 'b', 'y']
for i, color in enumerate(colors):
    plt.axhline(y=i, color=color, linestyle='--')

# 设置图像标题和坐标轴标签
plt.title('Scipy Dendrogram with Coloring')
plt.xlabel('Data Points')
plt.ylabel('Distance')

# 显示图像
plt.show()