"""
Showing clusters
================
"""

# %%
# A cluster is displayed as a subtree in the dendrogram. The :func:`~pdendro.attach_dendrogram`
# function can display clusters by specifying their root nodes. Let us assume that we have
# the following linkage matrix:

import scipy

X = [[0.0], [1.0], [3.0], [6.0]]
Z = scipy.cluster.hierarchy.linkage(X)

# %%
# The data points can be assigned to clusters using the :func:`scipy.cluster.hierarchy.fcluster`
# function, and their root nodes can be obtained using the :func:`scipy.cluster.hierarchy.leaders`
# function:

T = scipy.cluster.hierarchy.fcluster(Z, 1.5, criterion="distance")
L, _ = scipy.cluster.hierarchy.leaders(Z, T)

# %%
# Then, the clusters can be shown with root node markers and different colors:

import plotly.graph_objects as go
import plotly.io as pio

from pdendro import attach_dendrogram

fig = go.Figure()
attach_dendrogram(fig, Z, subtrees=L)
pio.show(fig)