heatmap-like plot, but for categorical variables in seaborn

by

You can use a discrete colormap and modify the colorbar, instead of using a legend.

value_to_int = {j:i for i,j in enumerate(pd.unique(df.values.ravel()))} # like you did
n = len(value_to_int)     
# discrete colormap (n samples from a given cmap)
cmap = sns.color_palette("Pastel2", n) 
ax = sns.heatmap(df.replace(value_to_int), cmap=cmap) 
# modify colorbar:
colorbar = ax.collections[0].colorbar 
r = colorbar.vmax - colorbar.vmin 
colorbar.set_ticks([colorbar.vmin + r / n * (0.5 + i) for i in range(n)])
colorbar.set_ticklabels(list(value_to_int.keys()))                                          
plt.show()

categorical seaborn heatmap

The colorbar part is adapted from this answer

HTH

More Related Contents:

  1. How to plot in multiple subplots
  2. Plotting with seaborn using the matplotlib object-oriented interface
  3. Move seaborn plot legend to a different position
  4. How can I make a barplot and a lineplot in the same seaborn plot with different Y axes nicely?
  5. How to display custom values on a bar plot
  6. How to change figuresize using seaborn factorplot
  7. How to format seaborn/matplotlib axis tick labels from number to thousands or Millions? (125,436 to 125.4K)
  8. How to plot and annotate grouped bars in seaborn / matplotlib
  9. Putting text in top left corner of matplotlib plot
  10. How do you plot a vertical line on a time series plot in Pandas?
  11. How can plot Artists be reused (Line2D)?
  12. Repeating x axis labels for all facets using FacetGrid in seaborn
  13. Stacked bar chart in Seaborn
  14. How can I group a stacked bar chart?
  15. Emulating deprecated seaborn distplots
  16. Improve subplot size/spacing with many subplots
  17. How to add percentages on top of grouped bars
  18. How to edit properties of whiskers, fliers, caps, etc. in Seaborn boxplot
  19. Plot multiple columns of pandas DataFrame using Seaborn
  20. Bar labels in matplotlib/Seaborn
  21. Remove seaborn lineplot legend title
  22. Normed histogram y-axis larger than 1
  23. How to express classes on the axis of a heatmap in Seaborn
  24. Seaborn tsplot does not show datetimes on x axis well
  25. How to plot a mean line on a kdeplot between 0 and the y value of the mean
  26. Python package to plot two heatmaps in one (split each square into two triangles)
  27. Matplotlib: avoiding overlapping datapoints in a “scatter/dot/beeswarm” plot
  28. Facetgrid change xlabels
  29. countplot() with frequencies
  30. How to change the color of the axis, ticks and labels

Leave a Comment