Skip to content

Pythoneo

Online How to Python stuff

How to create Seaborn Heatmap?

Posted on December 29, 2021March 2, 2023 By Luke K

Let’s learn how to create Seaborn Heatmap chart in Python.
Seaborn Python heatmap chart

You may need heatmap chart when you want to see fluctuations of two categories. Heatmap are useful for visualisation of two-dimentional data.

To create a heatmap you will need to import Python modules like: Seaborn, Matplotlib, Numpy and Pandas. It depends on your setup and the source of your data. I am going to generate the data so I don’t need Pandas which would be useful when you are taking the data from external source like csv file.

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

This is how I’m generating the data.

my_random_data = np.random.rand(4, 4)

It’s just to have data to work with.

See also  How to generate random integers in range with Numpy?

I’m going to define the data sets to show in the heatmap, create subplots and set the size of the chart.

f, ax = plt.subplots(figsize=(9, 6))

Sns heatmap

Let’s jump to the core heatmap definition.

ax = sns.heatmap(my_random_data, annot = True,
                 cmap = 'coolwarm', linecolor = 'black',
                 linewidths = 2, robust = True)

I’m going to save my heatmap as a png picture.

f.savefig('My heatmap.png')

To see the heatmap I need Matplotlib show method. This is not mandatory as I’ve already saved my Seaborn heatmap as a png picture.

plt.show()

My Seaborn heatmap is almost ready. It requires just some polishing. I need to name my chart and labels to improve the visibility.

ax.set_title('Heatmap by Pythoneo.com')
ax.set_xlabel('My X label')
ax.set_ylabel('My Y label')

The heatmap is ready.

See also  How to generate evenly spaced sample in Numpy?

Seaborn Python heatmap chart

Heatmap code

That’s the full code:

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

my_random_data = np.random.rand(4, 4)
f, ax = plt.subplots(figsize=(9, 6))
ax = sns.heatmap(my_random_data, annot = True,
                 cmap = 'coolwarm', linecolor = 'black',
                 linewidths = 2, robust = True)
ax.set_title('Heatmap by Pythoneo.com')
ax.set_xlabel('My X label')
ax.set_ylabel('My Y label')
f.savefig('My heatmap.png')
plt.show()
matplotlib, numpy, Pandas, Seaborn Tags:chart, heatmap, picture

Post navigation

Previous Post: How to create a BarPlot in SeaBorn?
Next Post: How to uninstall Numpy?

Categories

  • bokeh (1)
  • datetime (3)
  • Django (5)
  • glob (1)
  • io (1)
  • json (1)
  • math (5)
  • matplotlib (10)
  • numpy (95)
  • OpenCV (1)
  • os (3)
  • Pandas (2)
  • paramiko (1)
  • pathlib (2)
  • Pillow (3)
  • Plotly (3)
  • Python (29)
  • random (7)
  • requests (1)
  • Scipy (4)
  • Seaborn (7)
  • shutil (1)
  • sqlite3 (1)
  • statistics (16)
  • sys (1)
  • Tkinter (9)
  • turtle (2)
  • Uncategorized (1)
  • urllib (1)
  • webbrowser (1)

RSS RSS

  • How to create violin plot using seaborn?
  • How To Use Colormaps In Matplotlib?
  • How to calculate bonds in Python
  • How to handle trigonometry in Python
  • How to Convert Int to Binary in Python?
  • How to fix ValueError: The truth value of an array with zero elements is ambiguous?
  • How to solve NameError: name ‘numpy’ is not defined
  • How to insert seaborn lineplot?
  • How to Find the Length of an Array in Python?
  • How to reset secret key in Django

Tags

arithmetic mean array axis button calculations chart conversion copy count counter data type dictionary dimension draw error files fill float generate grid GUI image index integer list matrix max mean median min normal distribution plot random reshape rotate round size standard deviation string sum test text time variance zero

Copyright © 2023 Pythoneo.

Powered by PressBook WordPress theme

We use cookies on our website to give you the most relevant experience by remembering your preferences and repeat visits. By clicking “Accept”, you consent to the use of ALL the cookies.
Cookie settingsACCEPT
Manage consent

Privacy Overview

This website uses cookies to improve your experience while you navigate through the website. Out of these, the cookies that are categorized as necessary are stored on your browser as they are essential for the working of basic functionalities of the website. We also use third-party cookies that help us analyze and understand how you use this website. These cookies will be stored in your browser only with your consent. You also have the option to opt-out of these cookies. But opting out of some of these cookies may affect your browsing experience.
Necessary
Always Enabled
Necessary cookies are absolutely essential for the website to function properly. This category only includes cookies that ensures basic functionalities and security features of the website. These cookies do not store any personal information.
Non-necessary
Any cookies that may not be particularly necessary for the website to function and is used specifically to collect user personal data via analytics, ads, other embedded contents are termed as non-necessary cookies. It is mandatory to procure user consent prior to running these cookies on your website.
SAVE & ACCEPT