如何解决PairGrid用scatter_kws时Seaborn residplot忽略hue的问题

这个问题我之前也碰到过，本质是因为sns.residplot本身并不是为PairGrid的hue分组场景设计的，当你直接传递scatter_kws参数时，它会覆盖PairGrid自动为hue分配的颜色设置，导致所有散点颜色一致。下面是具体的复现和解决方法：

先复现你的问题 #

用你提供的数据，写出导致问题的代码：

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

# 生成随机数据
categories = pd.DataFrame(np.array(["p","q"])[np.random.randint(0,2,size=[100,1])],columns=["cat"],dtype="object")
data = pd.DataFrame(np.random.rand(100,2),columns=list('AB'))
df = pd.concat([data,categories],axis=1)

# 问题代码：scatter_kws导致hue颜色失效
g = sns.PairGrid(df, hue="cat")
g.map_offdiag(sns.residplot, scatter_kws={"s": 50, "alpha": 0.7})
g.add_legend()
plt.show()

运行这段代码你会发现，两个类别的散点颜色完全一样，hue参数没起作用。

解决方案1：自定义包装函数继承PairGrid的hue颜色 #

我们可以写一个简单的包装函数，接收PairGrid传递的hue对应颜色，再把它传给residplot的scatter_kws，这样既能保留你想要的散点样式，又能让hue颜色生效：

def residplot_with_hue(x, y, hue=None, color=None, **kwargs):
    # 提取scatter_kws，添加当前hue对应的颜色
    scatter_kws = kwargs.pop("scatter_kws", {})
    scatter_kws["color"] = color
    # 调用residplot
    sns.residplot(x=x, y=y, scatter_kws=scatter_kws, **kwargs)

# 使用自定义函数绘制
g = sns.PairGrid(df, hue="cat")
g.map_offdiag(residplot_with_hue, scatter_kws={"s": 50, "alpha": 0.7})
g.add_legend()
plt.show()

这个函数会自动获取PairGrid为每个hue分组分配的颜色，然后传递给散点，完美解决冲突。

解决方案2：手动映射调色板（更灵活） #

如果你想自定义hue对应的颜色，可以先获取Seaborn的调色板，然后手动给每个数据点分配颜色：

# 定义hue到颜色的映射，这里用默认调色板的前两个颜色
palette = sns.color_palette()
color_map = {"p": palette[0], "q": palette[1]}

def residplot_with_hue(x, y, data, **kwargs):
    # 获取当前子图的hue数据，生成颜色列表
    hue_colors = [color_map[h] for h in data["cat"]]
    scatter_kws = kwargs.pop("scatter_kws", {})
    scatter_kws["c"] = hue_colors
    sns.residplot(x=x, y=y, data=data, scatter_kws=scatter_kws, **kwargs)

g = sns.PairGrid(df, hue="cat")
g.map_offdiag(residplot_with_hue, scatter_kws={"s": 50, "alpha": 0.7})
g.add_legend()
plt.show()

这种方法适合你需要自定义颜色的场景，比如品牌色、特定配色方案。

为什么原来的方法失效？ #

简单来说，residplot本身不支持hue参数（早期版本完全不支持，新版本可能有限支持），PairGrid的hue映射是通过在map函数中为每个分组传递不同的颜色实现的，但当你设置scatter_kws时，会直接覆盖这个颜色设置，导致所有散点使用同一颜色。

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