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

Question

我有一个399行的熊猫数据框，正在与我一起制作热图，并且正在使用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。 有办法吗？

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

Answer 1

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

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