在 seaborn catplot 上放置省略号

Question

我有一个seaborn.catplot ，看起来像这样：

我想做的是用以下规则突出图表中的差异：

• 如果 AB > 4，将其着色为绿色
• 如果 AB < -1，将其涂成红色
• 如果 AB = <2= 且 >=0，将其涂成蓝色

我正在寻找类似于下图的东西：

我这里有一个MRE ：

# Stack Overflow Example
import numpy as np, pandas as pd, seaborn as sns
from random import choice
from string import ascii_lowercase, digits

chars = ascii_lowercase + digits
lst = [''.join(choice(chars) for _ in range(2)) for _ in range(100)]

np.random.seed(8)
t = pd.DataFrame(
    {
    'Key': [''.join(choice(chars) for _ in range(2)) for _ in range(5)]*2,
    'Value': np.random.uniform(low=1, high=10, size=(10,)), 
    'Type': ['A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'B']
    }
)

ax = sns.catplot(data=t, x='Value', y='Key', hue='Type', palette="dark").set(title="Stack Overflow Help Me")
plt.show()

我相信需要在兴趣点周围绘制一个省略号，我已经研究了一些问题：

• 使用 matplotlib 在散点图中创建置信椭圆
• plot seaborn 散点图中的椭圆 plot

但似乎没有人特别针对catplot或自定义颜色和规则来执行此操作。

我怎样才能用我的玩具示例获得预期的结果？

Answer 1

您可以在 A 和 B 的中点周围创建椭圆，使用 A 和 B 之间的距离，增加一些填充作为宽度。 高度应略小于 1。

要获得完整的轮廓和透明的内部颜色，可以使用to_rgba() 。 将 zorder 设置为较低的数字会将椭圆放在散点后面。

sns.scatterplot是sns.catplot的轴级等价物，当只有一个子图时更容易使用。

制作pd.Categorical类型的Key列给出了 y 位置和 label 之间的固定关系。

import matplotlib.pyplot as plt
from matplotlib.patches import Ellipse
from matplotlib.colors import to_rgba
import seaborn as sns
import pandas as pd
import numpy as np
from string import ascii_lowercase, digits

chars = ascii_lowercase + digits
num = 9
df = pd.DataFrame({'Key': [''.join(np.random.choice([*chars], 2)) for _ in range(num)] * 2,
                   'Value': np.random.uniform(low=1, high=10, size=2 * num),
                   'Type': np.repeat(['A', 'B'], num)})
df['Key'] = pd.Categorical(df['Key'])  # make the key categorical for a consistent ordering

sns.set_style('white')
ax = sns.scatterplot(data=df, x='Value', y='Key', hue='Type', palette="dark")
df_grouped = df.groupby(['Key', 'Type'])['Value'].mean().unstack()
for y_pos, y_label in enumerate(df['Key'].cat.categories):
    A = df_grouped.loc[y_label, 'A']
    B = df_grouped.loc[y_label, 'B']
    dif = A - B
    color = 'limegreen' if dif > 4 else 'crimson' if dif < -1 else 'dodgerblue' if 0 <= dif < 2 else None
    if color is not None:
        ell = Ellipse(xy=((A + B) / 2, y_pos), width=abs(dif) + 0.8, height=0.8,
                      fc=to_rgba(color, 0.1), lw=1, ec=color, zorder=0)
        ax.add_patch(ell)
plt.tight_layout()
plt.show()