使用 yticklabels = "auto" 从 sns.heatmap 获取行名

Question

我有一个包含 399 行的 pandas 数据框，我正在使用它制作热图，并且正在使用 yticklabels = 'auto' 功能。

当我使用get_yticklabels() 选项时：

sns.heatmap(df, ax = ax, robust = True, yticklabels = "auto").get_yticklabels()

这会返回 <a list of 157 Text yticklabel objects>.

我尝试遍历列表，但我只是获取 <class 'matplotlib.text.Text'> 类的对象，其中条目将类似于 Text(0, 0.5, 'hello')。我无法访问该对象中的数据来提取字符串hello。 有没有办法做到这一点？

如何获取热图中使用的行列表而不是对象？ 我的目标是仅提取 seaborn 热图中的数据框行，以进行进一步的下游分析。这就是为什么我需要这些值的列表。

Answer 1

labels = [t.get_text() for t in ax.get_yticklabels()]

将是 y 轴刻度标签的字符串列表。