matplotlib: 如何设置生成的图片比例?

为了去除图片的边缘空白部分，我使用了

fig = plt.figure()
ax = fig.add_subplot(111)
extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
plt.savefig('test.png', dpi = 200, bbox_inches = extent)

然后默认生成的图片宽高比例都是 31:24 的。

测试了一下直接 show

plt.show()

show出来的图片宽高比例是1：1的。

该如何设置图片的比例？