Lines to separate groups in seaborn heatmap

python matplotlib heatmap seaborn
T. Carson picture T. Carson · Sep 6, 2016 · Viewed 9k times · Source

I am plotting data as a Seaborn heatmap in Python. My data is intrinsically grouped into categories, and I'd like to have lines on the plot to indicate where the groups lie on the map. As a simple example, suppose I wanted to modify this plot from the documentation...

import seaborn as sns; sns.set()
flights = sns.load_dataset("flights")
flights = flights.pivot("month", "year", "passengers")
ax = sns.heatmap(flights, cbar=False)

enter image description here

Where I wanted to emphasize the comparisons between quarters of the year by making a plot like the one below; how would I do that?

enter image description here

Answer

Stop harming Monica picture Stop harming Monica · Sep 6, 2016

You want ax.hlines:

ax.hlines([3, 6, 9], *ax.get_xlim())

Related questions

seaborn heatmap y-axis reverse order

Have a look at this heatmap found in the seaborn heatmap documentation. Right now the y-axis starts with 9 at the bottom, and ends with 0 on top. Is there a way to turn this around, i.e. start with 0 at bottom …

Read More
Set Max value for color bar on seaborn heatmap

I need to set the max value on the seaborn heatmap cbar to 2. I've tried: cbar_kws = { 'ticks' : [0, 2] } sns.heatmap(tiles, robust=True, fmt="f", cmap= 'RdBu_r', cbar_kws = cbar_kws) But this doesn't work and the documentation isn't …

Read More
One colorbar for seaborn heatmaps in subplot

Here is an example that shows a colorbar for each subplot: import seaborn as sns import matplotlib.pyplot as plt import pandas as pd import numpy as np df = pd.DataFrame(np.random.random((10,10,))) fig,axn = plt.subplots(2, 2, sharex=True, …

Read More