带有分类/误分类实例数量的混淆矩阵（Python/Matplotlib）

Question

我正在用matplotlib绘制一个混淆矩阵，代码如下：

from numpy import *
import matplotlib.pyplot as plt
from pylab import *

conf_arr = [[33,2,0,0,0,0,0,0,0,1,3], [3,31,0,0,0,0,0,0,0,0,0], [0,4,41,0,0,0,0,0,0,0,1], [0,1,0,30,0,6,0,0,0,0,1], [0,0,0,0,38,10,0,0,0,0,0], [0,0,0,3,1,39,0,0,0,0,4], [0,2,2,0,4,1,31,0,0,0,2], [0,1,0,0,0,0,0,36,0,2,0], [0,0,0,0,0,0,1,5,37,5,1], [3,0,0,0,0,0,0,0,0,39,0], [0,0,0,0,0,0,0,0,0,0,38] ]

norm_conf = []
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)

plt.clf()
fig = plt.figure()
ax = fig.add_subplot(111)
res = ax.imshow(array(norm_conf), cmap=cm.jet, interpolation='nearest')
cb = fig.colorbar(res)
savefig("confmat.png", format="png")

但是我希望混淆矩阵上能显示数字，就像这个图（右边的那个）。我该怎么把conf_arr上的数字显示在图上呢？

Answer 1

我能想到的唯一方法就是使用注解。试试这些代码：

for i,j in ((x,y) for x in xrange(len(conf_arr))
            for y in xrange(len(conf_arr[0]))):
    ax.annotate(str(conf_arr[i][j]),xy=(i,j))

在保存图形之前。这段代码会添加数字，不过具体怎么调整数字的大小就得靠你自己去摸索了。

Answer 2

你可以使用 text 这个功能在你的图表中添加任意文本。比如，把下面这些代码插入到你的代码中，就可以在图表上显示数字（注意，第一行和最后一行是你原来的代码，用来告诉你在哪里插入我的代码）：

res = ax.imshow(array(norm_conf), cmap=cm.jet, interpolation='nearest')
for i, cas in enumerate(conf_arr):
    for j, c in enumerate(cas):
        if c>0:
            plt.text(j-.2, i+.2, c, fontsize=14)
cb = fig.colorbar(res)