相关热图，y轴偏离一半

Question

我有一个这样的数据框：

import pandas as pd
df = pd.DataFrame({'a':[0,9,8,7],
                   'b':[1,2,3,4], 
                   'c':[4,0,6,7]})

，我想将相关性df.corr()绘制为热图。

plt.imshow(df.corr())  

我想要的东西

但是当我在刻度上添加标签

plt.xticks(range(df.shape[1]), df.columns, rotation=90)  
plt.yticks(range(df.shape[1]), df.columns)  

第一行中所有单元格的上半部分以及底部行中所有单元格的下半部分被切除。

我特别困惑，即使我在两个轴上旋转标签，它也只在一个轴上发生。

如何获得完整的单元格和列标签？

感谢您指出现有答案。那里的答案针对seaborn，因此适用于matplotlib：

yy = plt.ylim()
plt.xticks(range(df.shape[1]), df.columns, rotation=90)
plt.yticks(range(df.shape[1]), df.columns)  
plt.ylim(yy)