使用matplotlib pcolor时多余的白色

Question

我正在尝试使用matplotlib将101x145矩阵绘制为热图。但是，情节的顶部和右侧有多余的白色。我该如何摆脱它？

考虑以下示例：

import cPickle as pickle
import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()
ax.pcolor(np.zeros((101,145)))
plt.show()

Answer 1

fig.tight_layout 做你想要的。它考虑了所有子图，轴标签，轴可刻度等，并尝试拉伸绘制的内容，以便尽可能减少空白空间。您还可以在图坐标中决定填充和框，其中tight_layout应该在其中工作。

Here就是一些例子。

警告：tight_layout仅适用于不包含Axes个对象的子图