0

我是使用python的初学者，遇到了一个问题，希望您能帮助我：

您可以在下面看到一个示例代码，用于绘制一个矩阵，其中 y 轴显示在左侧，x 轴显示在顶部。我想要的是在每列下方底部的 x 轴上具有该列的总和，对于每一行右侧的 y 轴是该行的总和。这意味着对于第一行，我想要数字 39 作为 y-label，对于第一列，我想要 39 作为 x-label。

我希望有人可以帮助我解决这个问题

import numpy as np
import matplotlib.pyplot as plt

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)

fig = plt.figure()
plt.clf()
ax = fig.add_subplot(111)
ax.set_aspect(1)
res = ax.imshow(np.array(norm_conf), cmap=plt.cm.OrRd, 
                interpolation='nearest')

width = len(conf_arr)
height = len(conf_arr[0])

for x in xrange(width):
    for y in xrange(height):
        ax.annotate(str(conf_arr[x][y]), xy=(y, x), 
                    horizontalalignment='center',
                    verticalalignment='center')

ax.xaxis.tick_top()
cb = fig.colorbar(res)
alphabet = 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'
plt.ylabel('True Landform (value explanation in info)')
plt.xticks(range(width), alphabet[:width])
plt.yticks(range(height), alphabet[:height])
plt.savefig('confusion_matrix.png', format='png')
plt.show()
4

1 回答 1

0

您可以使用以下命令对第 #i 行求和：

sum(conf_arr[i])

请记住，在 python 中，计数从 0 开始（而不是像 Matlab 那样从 1 开始）。

要反转矩阵，您可以使用

zip(*conf_arr)

然后您可以对第#i 行求和（实际上是第#i 列）。

于 2013-11-07T09:36:23.200 回答