如何在Seaborn分类轴蜂群图中标注特定点并使用其他列标签

问题背景 #

需要在基于Iris数据集的Seaborn分类swarm图中，为sepal width (cm)大于3.6的点添加标注，标注内容为对应行的sepal length (cm)值，同时解决以下问题：

• 原代码抛出TypeError: 'FacetGrid' object is not callable错误
• 分类轴场景下无法直接套用数值轴的标注逻辑
• 标注文本存在重叠问题

错误原因 #

sns.catplot()返回的是FacetGrid对象，而非Matplotlib的Axes对象，直接调用a_viz.text()会触发类型错误。此外，分类y轴的标签是字符串，需要先映射为对应的数值坐标才能正确定位标注。

解决方案步骤 #

• 获取FacetGrid对应的Axes对象：通过a_viz.axes[0][0]获取绘图的坐标轴实例
• 映射分类y轴到数值坐标：利用Axes的get_yticklabels()和get_yticks()方法，建立类别名称到数值位置的映射
• 使用目标列值作为标注文本：替换固定字符串'outlier'为dfiris['sepal length (cm)']的对应值
• 解决标注重叠：使用adjustText库自动调整标签位置，避免重叠

完整代码 #

import pandas as pd
import numpy as np
from sklearn.datasets import load_iris
import seaborn as sns
from adjustText import adjust_text

# 加载并处理数据集
dfiris = load_iris()
dfiris = pd.DataFrame(data=dfiris.data, columns=dfiris.feature_names)
dfiris['name'] = np.where(dfiris['sepal width (cm)'] < 3, 'Amy', 'Bruce')
dfiris['name'] = np.where((dfiris.name != 'Amy') & (dfiris['petal length (cm)'] >= 3.4), 'Charles', dfiris.name)

# 创建分类swarm图
a_viz = sns.catplot(x='sepal width (cm)', y='name', kind='swarm', data=dfiris)
a_viz.fig.set_size_inches(5, 6)
a_viz.fig.subplots_adjust(top=0.81, right=0.86)

# 获取坐标轴实例
ax = a_viz.axes[0][0]

# 建立类别到y坐标的映射
y_labels = [label.get_text() for label in ax.get_yticklabels()]
y_pos_map = {label: pos for label, pos in zip(y_labels, ax.get_yticks())}

# 收集需要标注的文本对象
texts = []
for idx, row in dfiris.iterrows():
    if row['sepal width (cm)'] > 3.6:
        # 获取对应的y坐标
        y_pos = y_pos_map[row['name']]
        # 添加标注，文本为sepal length的值（保留1位小数）
        txt = ax.text(row['sepal width (cm)'], y_pos, 
                      f"{row['sepal length (cm)']:.1f}", 
                      horizontalalignment='left', 
                      size='medium', 
                      color='black')
        texts.append(txt)

# 自动调整标注位置，解决重叠
adjust_text(texts, ax=ax)

说明 #

• 使用adjustText前需要先安装：pip install adjustText
• 标注文本保留一位小数，可根据需求修改格式化字符串f"{...:.1f}"
• 分类轴的坐标映射确保标注能准确对应到每个类别的行

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