rich-heatmap 0.1.3

A simple rich renderable Heatmap.

rich_heatmap

A simple Heatmap renderable for projects using rich.

Installing

Install with pip or your favorite package manager:

python -m pip install rich_heatmap

uv add rich_heatmap

Usage

Create HeatmapCell objects containing your data:

for row, col in itertools.product(range(30), range(20)):
    value = noise.snoise2(col / 10, row / 15)
    cells.append(heatmap.HeatmapCell(row, col, value, None, f"{value:0.1f}"))

Values will be normalized into [0, 1] and then mapped to a color using a colormap function:

EMBER = matplotlib.colormaps['cmr.ember']
def colormap(value: float) -> tuple[float, float, float]:
	rgba = EMBER(value)
	return rich.color.Color.from_rgb(
		255 * rgba[0],
		255 * rgba[1],
		255 * rgba[2],
	)

And then print a Heatmap containing those cells:

print(heatmap.Heatmap(cells=cells, colormap=colormap))

Row and column values will be sorted before printing, so you can accumulate a stream of data and feed it in without any more trouble:

If you don't provide text, the library will produce rectangular cells: