我想使用matplotlib创建混淆矩阵的可视化。以下方法的参数为类标签(字母表),分类结果作为列表的列表(conf_arr)和输出文件名。到目前为止,我对结果感到非常满意,但还有一个问题:我无法将轴刻度标签居中对齐。如果我传递下面所示的extent参数给imshow方法,则网格线与我希望对齐的位置相对应。如果我注释掉它,则网格线会错位,但标签会在我希望的位置上。我认为我需要一种方法将刻度标签移动到相关刻度和下一个刻度之间,但我不知道这是否可能以及如何做到。总之,我希望像右图那样放置刻度标签,但像左图那样具有网格/刻度。![在此输入图像描述](https://istack.dev59.com/Kuy6w.webp)
def create_confusion_matrix(alphabet, conf_arr, outputname):
norm_conf = []
width = len(conf_arr)
height = len(conf_arr[0])
for i in conf_arr:
a = 0
tmp_arr = []
a = sum(i, 0)
for j in i:
tmp_arr.append(float(j)/float(a))
norm_conf.append(tmp_arr)
fig = plt.figure(figsize=(14,14))
#fig = plt.figure()
plt.clf()
ax = fig.add_subplot(111)
ax.set_aspect(1)
ax.grid(which='major')
res = ax.imshow(np.array(norm_conf), cmap=plt.cm.binary,
interpolation='none', aspect='1', vmax=1,
##Commenting out this line sets labels correctly,
##but the grid is off
extent=[0, width, height, 0]
)
divider = make_axes_locatable(ax)
cax = divider.append_axes("right", size="5%", pad=0.2)
cb = fig.colorbar(res, cax=cax)
#Axes
ax.set_xticks(range(width))
ax.set_xticklabels(alphabet, rotation='vertical')
ax.xaxis.labelpad = 0.5
ax.set_yticks(range(height))
ax.set_yticklabels(alphabet, rotation='horizontal')
#plt.tight_layout()
plt.savefig(outputname, format='png')
生成的图片如下所示:
![在此输入图像描述](https://istack.dev59.com/Kuy6w.webp)