%matplotlib widget
import matplotlib.pyplot as plt
import numpy as np
from mpl_interactions import heatmap_slicer
Sometimes I find myself wanting to compare horizontal or vertical slices across two different heatmaps with the same shape. The function {func}.heatmap_slicer
makes this easy and should work for any number of heatmaps from 1 to many (likely not all the way $\infty$ though).
The most important options to play with are:
slices
: "both"
, "vertical"
, or "horizontal"
interaction_type
: "move"
or "click"
x = np.linspace(0, np.pi, 100)
y = np.linspace(0, 10, 200)
X, Y = np.meshgrid(x, y)
data1 = np.sin(X) + np.exp(np.cos(Y))
data2 = np.cos(X) + np.exp(np.sin(Y))
fig, axes = heatmap_slicer(
x,
y,
(data1, data2),
slices="both",
heatmap_names=("dataset 1", "dataset 2"),
labels=("Some wild X variable", "Y axis"),
interaction_type="move",
)