"""
Grouped heatmap
===============


"""

# %%
import numpy as np
import marsilea as ma

data = np.random.randint(0, 10, (10, 10))
groups = np.random.choice(["A", "B", "C"], 10)

# %%

h = ma.Heatmap(data)
h.add_dendrogram("right")
h.group_rows(groups, order=["A", "B", "C"])
h.add_left(ma.plotter.Chunk(["A", "B", "C"], fill_colors=["#F05454", "#F0F0F0", "#54F0F0"]))
h.add_legends("left")
h.add_title("Grouped heatmap")
h.render()