pythonの階層クラスタリングにおいて、任意の階層でラベルづけすることは可能ですか？

Question

初心者の質問で申し訳ございません。
そもそも機械学習を理解していない質問になってしまうかもしれないのですが、
みなさまのお力を拝借できますと幸いです。

階層クラスタリングのラベル付けに苦戦しています。

12000行×13列のデータフレームをlinkage()でクラスタリングをかけました。
（中身は0/1回答フラグです）（そもそも12000が多すぎるのは分かっているのですが、できればこのままでいきたいです）
コードは下記です。

linkage_result = linkage(df, method='ward', metric='euclidean')
dendrogram(linkage_result)
t = 0.7*max(linkage_result[:,2])
c = fcluster(linkage_result, t, criterion = "distance")

　

この際、ラベルが1 or 2で出てきてしまうのですが、
下記画像のようにピンクの線（任意の階層）で切って、その分岐数のラベルを付与することは可能でしょうか。

!![イメージ説明

Answer 1

ぴったりそのままのことはできませんが、dendrogramのオプションで途中で切ること自体は可能です。

import matplotlib.pyplot as plt
from scipy.cluster.hierarchy import linkage, dendrogram
from sklearn.datasets import load_iris

iris = load_iris()
z = linkage(iris.data)
fig, axes = plt.subplots(1, 3, figsize=(10, 5))

ps = [30, 10, 4]
modes = [None, "lastp", "level"]

for ax,p,mode in zip(axes, ps, modes):
    dendrogram(z, labels=iris.target,
               p=p, truncate_mode=mode,
               ax=ax)
plt.savefig("result.png")

scipy.cluster.hierarchy.dendrogram — SciPy v1.2.0 Reference Guide

Comments

[aooo]

hayataka2049さん、ご回答ありがとうございます。
いただいたコードを紐解くにあたって、dendrogramのガイドも読みまして、
ああやってパラメーターを確認するんだ！と
自分で調べるやり方も学ぶことができました。
大変感謝です。ありがとうございました。