减少pcolor matplotlib图中的空白

Question

我目前正在使用matplotlib pcolor绘制旋转表，但是我的轴没有像我希望的那样工作。我用以下方法绘制表格：

pyl.pcolor(pivot_99)
pyl.colorbar()
pyp.suptitle('Pivot 99', fontsize=14)
pyp.xlabel('Site Number', fontsize=12)
pyp.ylabel('Site Number', fontsize=12)
pyp.yticks(arange(len(pivot_99.index)),pivot_99.index)
pyp.xticks(arange(len(pivot_99.columns)),pivot_99.columns, rotation = 90)
pyl.savefig('Pivot 99.png')
pyl.show()

其中pivot_99是一个由0到1之间的数字组成的数据，索引和列的长度总是相同的，但在不同的年份是不同的值。当我绘制pivot_99时，它看起来很好：

​

然而，当我绘制2000热图时，顶部和右边都有额外的空间。

​

​

有人知道为什么会发生这种情况吗?或者我怎样才能修复这个图形，这样只有颜色的空间才会显示出来？谢谢

Answer 1

尝试使用set_xlim和set_ylim，如本例中所示：

import matplotlib.pyplot as pyp

pyp.gca().set_xlim((1,10))
pyp.gca().set_ylim((1,12))
pyp.plot([1,2],[1,2])
pyp.show()

您可以尝试使用xlim=ylim=558，我可以测试它，因为我没有您的数据：

Answer 2

或者，您也可以使用prettyplotlib，它将自动为您设置xlim，ylim，移除您的刺和蜱，并给您一个合理的颜色条。

import prettyplotlib as ppl
ppl.pcolormesh(pivot_99)

完全披露:我是prettyplotlib作者:)