H=dendrogramZ2改变谱系图不同聚类的颜色
要改变谱系图中不同聚类的颜色,可以使用scipy.cluster.hierarchy.dendrogram函数的link_color_func参数。link_color_func是一个函数,它接受聚类的索引(从0开始)作为输入,并返回相应聚类的颜色。
下面是一个示例代码,演示如何使用link_color_func来改变谱系图中不同聚类的颜色:
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, 'ward')
# 定义颜色函数
def get_color(index):
colors = ['r', 'g', 'b', 'c', 'm', 'y', 'k']
return colors[index % len(colors)]
# 绘制谱系图
plt.figure(figsize=(10, 5))
dendrogram(Z, link_color_func=lambda x: get_color(x))
plt.show()
在这个例子中,我们生成了一个包含10个样本的随机数据集。然后使用linkage函数进行层次聚类,并得到聚类结果Z。最后,我们定义了一个get_color函数,它根据聚类的索引返回相应的颜色。在绘制谱系图时,我们通过link_color_func参数传递了这个颜色函数,从而改变了谱系图中不同聚类的颜色。
你可以根据自己的需求自定义颜色函数,以实现不同的颜色映射
原文地址: https://www.cveoy.top/t/topic/i1NJ 著作权归作者所有。请勿转载和采集!