Heatmap

Skill Matrix Heatmap

Dark theme heatmap showing team skills assessment

Output
Skill Matrix Heatmap
Python
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import LinearSegmentedColormap
from matplotlib.patches import FancyBboxPatch

fig, ax = plt.subplots(figsize=(12, 9), facecolor='#ffffff')
ax.set_facecolor('#ffffff')

np.random.seed(42)
skills = ['Python', 'JavaScript', 'SQL', 'Docker', 'AWS', 'Git', 'Testing']
team = ['Alice', 'Bob', 'Charlie', 'Diana', 'Eve']
data = np.random.randint(1, 6, (len(team), len(skills)))

colors = ['#1e293b', '#fef3c7', '#fbbf24', '#0891b2', '#0c4a6e']
cmap = LinearSegmentedColormap.from_list('skill', colors, N=256)

cell_w, cell_h = 0.88, 0.82
for i in range(len(team)):
    for j in range(len(skills)):
        val = data[i, j]
        rect = FancyBboxPatch((j - cell_w/2, i - cell_h/2), cell_w, cell_h,
                               boxstyle="round,pad=0.02,rounding_size=0.12",
                               facecolor=cmap((val-1)/4), edgecolor='#e2e8f0', linewidth=1.5)
        ax.add_patch(rect)
        ax.text(j, i, str(val), ha='center', va='center', color='#1e293b', fontsize=11, fontweight='bold')

ax.set_xlim(-0.5, len(skills)-0.5)
ax.set_ylim(-0.5, len(team)-0.5)
ax.set_aspect('equal')
ax.invert_yaxis()
ax.set_xticks(range(len(skills)))
ax.set_yticks(range(len(team)))
ax.set_xticklabels(skills, rotation=45, ha='right', color='#64748b', fontsize=10, fontweight='500')
ax.set_yticklabels(team, color='#1e293b', fontsize=10, fontweight='500')

sm = plt.cm.ScalarMappable(cmap=cmap, norm=plt.Normalize(1, 5))
cbar = plt.colorbar(sm, ax=ax, shrink=0.8, pad=0.02)
cbar.set_label('Skill Level', color='#1e293b', fontsize=11)
cbar.outline.set_edgecolor('#e2e8f0')
plt.setp(plt.getp(cbar.ax.axes, 'yticklabels'), color='#64748b')

for spine in ax.spines.values(): spine.set_visible(False)
ax.tick_params(length=0)
ax.set_title('Team Skill Matrix', fontsize=18, color='#1e293b', fontweight='bold', pad=20)
plt.tight_layout()
plt.show()
Library

Matplotlib

Category

Heatmaps & Density

Did this help you?

Support PyLucid to keep it free & growing

Support