Seaborn – Heatmap Plot Visualization


Seaborn – Heatmap Plot Visualization

Table Of Contents:

  • What Is Heatmap Plot ?
  • Where To Use Heatmap Plot ?
  • Examples Of Heatmap Plot ?

(1) What Is Heatmap Plot ?

(2) Where To Use Heatmap Plot ?

(3) Examples Of Heatmap Plot ?

Example-1: Basic Heatmap for Correlation Matrix

import seaborn as sns
import matplotlib.pyplot as plt

# Load built-in dataset
tips = sns.load_dataset("tips")

# Keep only numeric columns
tips_numeric = tips.select_dtypes(include="number")

# Compute correlation matrix
corr = tips_numeric.corr()

# Create heatmap
sns.heatmap(corr, annot=True, cmap="coolwarm", fmt=".2f")

plt.title("Correlation Heatmap")
plt.show()

Example-2: Heatmap for Missing Values

import numpy as np
import pandas as pd

# Create a sample dataset with missing values
data = pd.DataFrame({
    "A": [1, 2, np.nan, 4, 5],
    "B": [5, np.nan, np.nan, 3, 2],
    "C": [3, 4, 2, 1, np.nan]
})

# Create heatmap to show missing values
sns.heatmap(data.isnull(), cmap="viridis", cbar=False)

plt.title("Missing Values Heatmap")
plt.show()

Example-3: Heatmap for Categorical Data (Pivot Table)

import seaborn as sns
import matplotlib.pyplot as plt

# Load built-in dataset
flights = sns.load_dataset("flights")

# Convert data into a pivot table (Use keyword arguments)
pivot = flights.pivot(index="month", columns="year", values="passengers")

# Create heatmap
sns.heatmap(pivot, cmap="Blues", annot=True, fmt="d")

plt.title("Monthly Flight Passengers Heatmap")
plt.show()

(4) Heatmap Vs Correlation Matrix ?

Leave a Reply

Your email address will not be published. Required fields are marked *