Discrete legend in seaborn heatmap plot

by

Well, there’s definitely more than one way to accomplish this. In this case, with only three colors needed, I would pick the colors myself by creating a LinearSegmentedColormap instead of generating them with cubehelix_palette. If there were enough colors to warrant using cubehelix_palette, I would define the segments on colormap using the boundaries option of the cbar_kws parameter. Either way, the ticks can be manually specified using set_ticks and set_ticklabels.

The following code sample demonstrates the manual creation of LinearSegmentedColormap, and includes comments on how to specify boundaries if using a cubehelix_palette instead.

import matplotlib.pyplot as plt
import pandas
import seaborn.apionly as sns
from matplotlib.colors import LinearSegmentedColormap

sns.set(font_scale=0.8)
dataFrame = pandas.read_csv('LUH2_trans_matrix.csv').set_index(['Unnamed: 0'])

# For only three colors, it's easier to choose them yourself.
# If you still really want to generate a colormap with cubehelix_palette instead,
# add a cbar_kws={"boundaries": linspace(-1, 1, 4)} to the heatmap invocation
# to have it generate a discrete colorbar instead of a continous one.
myColors = ((0.8, 0.0, 0.0, 1.0), (0.0, 0.8, 0.0, 1.0), (0.0, 0.0, 0.8, 1.0))
cmap = LinearSegmentedColormap.from_list('Custom', myColors, len(myColors))

ax = sns.heatmap(dataFrame, cmap=cmap, linewidths=.5, linecolor="lightgray")

# Manually specify colorbar labelling after it's been generated
colorbar = ax.collections[0].colorbar
colorbar.set_ticks([-0.667, 0, 0.667])
colorbar.set_ticklabels(['B', 'A', 'C'])

# X - Y axis labels
ax.set_ylabel('FROM')
ax.set_xlabel('TO')

# Only y-axis labels need their rotation set, x-axis labels already have a rotation of 0
_, labels = plt.yticks()
plt.setp(labels, rotation=0)

plt.show()

Heatmap using red, green, and blue as colors with a discrete colorbar

More Related Contents:

  1. How to have clusters of stacked bars with python (Pandas)
  2. Python matplotlib multiple bars
  3. plot different color for different categorical levels using matplotlib
  4. How to plot multiple Seaborn Jointplot in Subplot
  5. Color by Column Values in Matplotlib
  6. Plotting grouped data in same plot using Pandas
  7. How to add value labels on a bar chart
  8. How to add hovering annotations to a plot
  9. How to draw vertical lines on a given plot
  10. Plot a horizontal line on a given plot
  11. How to have clusters of stacked bars
  12. How to display custom values on a bar plot
  13. How to plot multiple dataframes in subplots
  14. Seaborn showing scientific notation in heatmap for 3-digit numbers
  15. Seaborn: countplot() with frequencies
  16. multi index plotting
  17. How do you plot a vertical line on a time series plot in Pandas?
  18. How to remove or hide x-axis labels from a seaborn / matplotlib plot
  19. Plotting time-series data with seaborn
  20. How to deal with NaN values when plotting a boxplot
  21. How can I group a stacked bar chart?
  22. How to create a grouped bar plot
  23. Using pandas crosstab to create a bar plot
  24. How to annotate a seaborn barplot with the aggregated value
  25. plot different color for different categorical levels
  26. Color a scatter plot by Column Values
  27. countplot() with frequencies
  28. Add image annotations to bar plots
  29. Seaborn multiple barplots
  30. Add density curve on the histogram

Leave a Comment