matplotlib绘制混淆矩阵模版及显示一半问题

别人的显示：

我的显示:

修改坐标显示后，使用matplotlib绘制混淆矩阵可能出现矩阵上下只显示一半的问题，看着别人显示没有问题，一度怀疑自己的人品，后来发现这是由于matplotlib版本问题。
修改版本不如一句代码方便，只需要在代码中加入坐标轴的范围就好。设置坐标轴的范围是我们经常使用，却又不注意的地方。

xlocations = np.arange(len(['圆形', '三角形', '方形']))
plt.xticks(xlocations,['圆形', '三角形', '方形'])
plt.yticks(xlocations,['圆形', '三角形', '方形'])
plt.ylim(len(labels)-0.5,-0.5)