Category bar on top of heatmap

Hi there,

I was wondering, if there is a efficient way to add a categorical bar on top of a heatmap. (Image attached green/yellow bar). Ideally, this would fit between the tick-labels and the heatmap. I’m not seeking the exact impl, but rather some pointers, as I seem to run in circles here.

Thanks in advance.