使用Seaborn kdeplot绘制多填充等高线子图时存在空白问题

嗨，我看你在绘制多个填充等高线子图时遇到了空白的问题，咱们一步步来排查和修复这个问题：

可能的问题根源 #

从你给出的代码片段来看，几个关键点可能导致空白：

1. 子图数量计算逻辑存疑：cols = int((len(dataset)-2)/2) 如果计算结果不符合预期（比如为0），会导致子图布局异常
2. KDE绘图代码不完整：你没写完sns.kdeplot的调用，大概率是没有正确传入对应子图的数据，也没指定绘制到目标轴上
3. 样式设置冗余：sns.set_style("white")放在循环里重复设置没必要，而且可能干扰绘图
4. 单张子图的迭代问题：如果最终只生成1个子图，plt.subplots返回的是单个轴对象而非数组，直接遍历会出错

修复后的完整代码 #

下面是调整后的可运行代码，我补充了关键逻辑并修复了潜在问题：

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

def kdeplots_figure(dataset):
    # 先验证数据集结构，确保有足够的元素生成子图
    num_pairs = (len(dataset) - 2) // 2
    if num_pairs <= 0:
        raise ValueError("数据集元素不足，无法创建子图，请检查dataset结构")
    
    # 统一设置Seaborn样式，只需执行一次
    sns.set_style("white")
    
    # 创建子图，处理单张子图的特殊情况
    figure, axes = plt.subplots(1, num_pairs, figsize=(15, 6), sharey=True)
    if num_pairs == 1:
        axes = [axes]  # 转成列表，保证遍历逻辑统一
    
    # 遍历每个子图，绘制KDE填充等高线
    for i, ax in enumerate(axes):
        # 按你的逻辑从dataset中提取对应两列数据，这里假设每两个数组对应一个子图
        data_idx = 2 + 2 * i
        x_data = dataset[data_idx].flatten()  # 转成一维数组，适配kdeplot输入
        y_data = dataset[data_idx + 1].flatten()
        
        # 绘制填充KDE，指定绘制到当前子图轴上
        sns.kdeplot(x=x_data, y=y_data, fill=True, ax=ax, cmap="viridis")
        
        # 设置子图标题和标签
        ax.set_title(f"Cluster {i}", fontsize=12)
        ax.set_xlabel("X 特征")
        # 因为sharey=True，只给第一个子图保留Y轴标签，避免重复
        if i == 0:
            ax.set_ylabel("Y 特征")
        else:
            ax.set_ylabel("")
    
    # 自动调整子图间距，避免标题、标签重叠
    plt.tight_layout()
    return figure

关键修复说明 #

• 合法性检查：先判断子图数量是否合理，避免无效的绘图请求
• 轴对象处理：兼容单张子图的情况，确保遍历逻辑不会出错
• 数据适配：将numpy数组展平为一维，符合Seaborn kdeplot的输入要求
• 明确绘图目标：调用kdeplot时指定ax=ax，确保每个图绘制到对应的子图上
• 布局优化：用plt.tight_layout()自动调整子图间距，提升可视化效果

额外排查建议 #

如果还是出现空白，你可以做这些检查：

• 打印x_data和y_data的非NaN值数量，确认数据是否有效（全NaN会导致无绘图内容）
• 验证num_pairs的计算是否和你的数据集结构匹配，确保每个子图对应的两列数据确实存在
• 检查dataset中的每个numpy数组维度是否一致，避免因维度不匹配导致绘图失败

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