我想用Pylab绘制一个混淆矩阵。沿水平轴的类标签很长,所以我想绘制它们垂直旋转。但是,我也希望将它们绘制在轴的顶部而不是下面。
此命令可以在底部绘制垂直标签:
pylab.imshow(confusion_matrix)
pylab.xticks(..., rotation='vertical')
并且此命令可以在顶部绘制水平标签而不旋转:
pylab.matshow(confusion_matrix)
但是找不到两者兼而有之。以下命令不起作用。
pylab.matshow(confusion_matrix)
pylab.xticks(..., rotation='vertical')
你能建议一种方法用垂直旋转在轴顶部用xticks绘制混淆矩阵吗?谢谢。
修改
谢谢Mark,谢谢你的帮助。通过更仔细地检查刻度属性,它让我走上正轨。与你的答案和我想要的答案的唯一区别是将这个想法应用于AxesImage,而不是绘图。经过调查,答案是:
im = pylab.matshow(confusion_matrix)
for label in im.axes.xaxis.get_ticklabels():
label.set_rotation(90)
im.figure.show()
对那些阅读...不要忘记show()!我忘了我需要刷新数字。见下面的输出。
Confusion matrix with vertical labels. http://up.stevetjoa.com/rotate_ticklabels.png
答案 0 :(得分:12)
如果我理解正确的话,这会让你接近。您可能需要用空格“填充”标签,以将它们移出x轴线。
from matplotlib import pylab
pylab.plot([0, 6], [0, 6])
pylab.xticks([1,2,3,4,5,6],('one','two','three','four','five','six'),rotation='vertical',verticalalignment='bottom')
编辑回应评论
如果您希望它们在顶部x轴上垂直旋转,请尝试:
pylab.plot([0, 6], [0, 6])
pylab.xticks([1,2,3,4,5,6],('one','two','three','four','five','six'))
for tick in pylab.gca().xaxis.iter_ticks():
tick[0].label2On = True
tick[0].label1On = False
tick[0].label2.set_rotation('vertical')
答案 1 :(得分:1)
阅读你的帖子,并亲自尝试,我发现了一个非常简单的方法在轴上设置刻度:
pylab.gca().tick_top()
干杯!