seaborn - 错误的混淆矩阵图

Question

使用 pandas 数据框绘制混淆矩阵时 y 轴两端被切一半？

这就是我得到的:

我使用了这里的代码How can I plot a confusion matrix?使用 pandas 数据框:

import seaborn as sn
import pandas as pd
import matplotlib.pyplot as plt

array = [[13,1,1,0,2,0],
    [3,9,6,0,1,0],
    [0,0,16,2,0,0],
    [0,0,0,13,0,0],
    [0,0,0,0,15,0],
    [0,0,1,0,0,15]]        
df_cm = pd.DataFrame(array, range(6),range(6))
#plt.figure(figsize = (10,7))
sn.set(font_scale=1.4)#for label size
sn.heatmap(df_cm, annot=True,annot_kws={"size": 16})# font size

Answer 1

我解决了这个问题，我认为this post解释了为什么会发生这种情况。

简单来说，matplotlib 3.1.1 破坏了seaborn 热图；您可以通过降级到matplotlib 3.1.0来解决。