如何用Seaborn制作结合分类与连续变量的热力图（按变量调透明度）

完全可以通过直接生成RGBA颜色矩阵来实现这个需求，绕过Seaborn和Matplotlib的默认热力图逻辑，手动控制每个单元格的色调（由分类决定）和透明度（由连续值控制）。以下是具体实现步骤和代码：

步骤说明 #

1. 为每个分类类别分配固定的RGB颜色
2. 将分类数据转换为对应的RGB颜色矩阵
3. 将连续数值归一化到0-1范围，作为透明度（alpha）通道
4. 合并RGB矩阵与alpha通道，生成RGBA格式的颜色矩阵
5. 使用Matplotlib直接绘制该RGBA矩阵，并添加对应的图例和颜色条

完整代码示例 #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Patch
import seaborn as sns

# ---------------------- 生成示例数据 ----------------------
np.random.seed(42)
# 10x10的分类数据（3个类别）
cat_data = np.random.randint(0, 3, size=(10, 10))
# 10x10的连续数据（范围0-100）
cont_data = np.random.randint(0, 101, size=(10, 10))

# ---------------------- 处理分类颜色 ----------------------
# 为每个分类分配固定RGB颜色（用Seaborn调色板取色）
cat_colors = sns.color_palette("Set2", n_colors=3)  # 3个类别对应3种颜色
# 将分类数据转换为RGB矩阵：shape (10,10,3)
rgb_matrix = np.array([[cat_colors[cat] for cat in row] for row in cat_data])

# ---------------------- 处理透明度（连续值） ----------------------
# 将连续值归一化到0-1范围，作为alpha通道
norm_cont = (cont_data - cont_data.min()) / (cont_data.max() - cont_data.min())
# 合并RGB与alpha，生成RGBA矩阵：shape (10,10,4)
rgba_matrix = np.dstack([rgb_matrix, norm_cont])

# ---------------------- 绘制热力图 ----------------------
fig, ax = plt.subplots(figsize=(8, 7))

# 直接绘制RGBA矩阵
im = ax.imshow(rgba_matrix, aspect='auto', interpolation='nearest')

# ---------------------- 添加分类图例 ----------------------
legend_elements = [Patch(facecolor=cat_colors[i], label=f"类别 {i}") for i in range(3)]
ax.legend(handles=legend_elements, bbox_to_anchor=(1.25, 1), loc='upper left')

# ---------------------- 添加连续值透明度的颜色条 ----------------------
from matplotlib.cm import ScalarMappable
from matplotlib.colors import Normalize

sm = ScalarMappable(norm=Normalize(vmin=cont_data.min(), vmax=cont_data.max()), cmap='gray')
sm.set_array([])  # 只需要颜色条，不需要绑定数据
cbar = fig.colorbar(sm, ax=ax, orientation='vertical', pad=0.2)
cbar.set_label('连续数值（透明度对应）')

# ---------------------- 调整布局与显示 ----------------------
ax.set_xticks(np.arange(cat_data.shape[1]))
ax.set_yticks(np.arange(cat_data.shape[0]))
plt.tight_layout()
plt.show()

关键细节解释 #

• RGBA矩阵构造：每个单元格的颜色由(R,G,B,A)四个值控制，其中R/G/B来自分类对应的固定颜色，A（透明度）由归一化后的连续值决定（值越大越不透明）
• 图例与颜色条分离：分类的颜色用补丁（Patch）做图例，连续值的透明度用单独的灰度颜色条表示，清晰区分两个维度的信息
• 绕过默认逻辑：直接使用imshow绘制自定义RGBA矩阵，完全不受Seaborn热力图的默认颜色映射限制，实现100%自定义控制

内容的提问来源于stack exchange，提问作者BathtubBear