python - Python 混淆矩阵中数字的位置不合适

Question

我有这样的代码:

plt.figure(figsize=(8,5))
confusion_matrix = pd.crosstab(y_test, predictions, rownames=["Observed"], colnames=["Anticipated"])
sns.heatmap(confusion_matrix, annot=True, fmt= "d")
plt.show()

尽管如此，每个方 block 内的数字并不位于每个方 block 的中心，如下所示。如何更改此代码以使数字位于每个方 block 的中心位置？

Answer 1

如this post on datascience.stackexchange.com中所示，matplotlib 3.1.1 破坏了 sns.heatmap()。那里的答案建议降级到 matplotlib 3.1.0。但我已经在我的机器上安装了 3.1.2 并且它可以工作。因此您现在也许可以升级。

y_test = np.array(
    [
        "foo", "foo", "foo", "foo",
        "bar", "bar", "bar", "bar",
        "foo", "foo", "foo"
    ], dtype=object
)
predictions = np.array(
    [
        "one", "one", "one", "two",
        "one", "one", "one", "two",
        "two", "two", "one"
    ], dtype=object
)

plt.figure(figsize=(8, 5))
confusion_matrix = pd.crosstab(
    y_test, predictions, rownames=["Observed"], colnames=["Anticipated"]
)
sns.heatmap(confusion_matrix, annot=True, fmt="d")
plt.show()

这是带有相关包的pip list的结果。

Package            Version   Location
------------------ --------- -------------------------------------
matplotlib         3.1.2
pandas             0.25.1
seaborn            0.9.0