"""
Nested subtrees
===============
"""

# %%
# A subtree and a contained subtree can be given together to the `subtrees` parameter.

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

from pdendro import attach_dendrogram

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

fig = go.Figure()
attach_dendrogram(fig, Z, subtrees={5, 4})
pio.show(fig)