我正在遵循文档中给出的示例,该示例解释了如何使用Iris数据帧绘制分层聚类图。
在这个例子中,我们可以传递一个参数p,它将剪切图表,对标签进行分组:
plot_dendrogram(model, truncate_mode='level', p=3)
然后在运行算法后,我们有 2X 标签,然后我放p = 2
,在树状图上只有 X/3 叶子。
我想要的是拥有树状图函数在p=2
时绘制的数据框,该数据框根据层次结构切割标签,但我找不到如何获得它。
plot_dendrogram
返回用于绘制树状图的值的字典。请参阅文档中的"退货"部分: https://docs.scipy.org/doc/scipy/reference/generated/scipy.cluster.hierarchy.dendrogram.html#scipy.cluster.hierarchy.dendrogram
尝试:
d = plot_dendrogram(model, truncate_mode='level', p=3)
并检查d
.
如果您只对数据感兴趣并且明确不希望它绘制图形,plot_dendrogram
可以no_plot=True
接受参数。