Heatmap Slicer#
%matplotlib widget
import matplotlib.pyplot as plt
import numpy as np
from mpl_interactions import heatmap_slicer
Comparing heatmaps#
Sometimes I find myself wanting to compare horizontal or vertical slices across two different heatmaps with the same shape. The function 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",
)