Pyplot 在热图上 Y 轴的指定值上绘制水平线

Question

我目前有一个数据框，df：

In  [1]: df
Out [1]:    

   one        two
   0.5        4.62
   1.0       12.23        
   1.5       17.53          
   2.0       20.32        
   2.5       19.37
   3.0       17.77
   3.5        9.04

我已经尝试用 y 轴上的值为 2 的水平线绘制热图，但是它改为在热图的第二条上绘制线：

In  [2]: plt.figure(figsize=(30,7))
         plt.title('Test')
         ax = sns.heatmap(data=df, annot=True, )
         plt.xlabel('Test')
         ax.invert_yaxis()
         ax.axhline(2, ls='--')

实际输出

我希望能够在 y 轴上的指定值处绘制一条水平线，以便输入“2”将给出以下图表（这是通过替换"ax.axhline(3.5, ls='--')"在我当前的代码中）：

期望的输出

Answer 1

看起来情节只是一个图像。 因此，在坐标“2”处绘制一条线会将这条线放在第二行像素的上方。 要以所需值绘制线条，您可以搜索适当的像素行，例如使用

value = 2
index = np.abs(df['one'] - value).argmin()
ax.axhline(index + .5, ls='--')