根据轴尺寸调整图形尺寸？

Question

假设我用matplotlib创建了一个简单的plot

import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)

plot stuff etc .. . 

完成绘图程序后，我想调整图形的大小（但保持所有纵横比相同），以便将ax的总高度设置为一个常量值，即3英寸。 有没有一个方便快捷的方法来存档？在

编辑：我知道如何更改图形大小，但这里我想根据轴的大小调整它。在

Answer 1

以下是以英寸为单位设置图形高度的函数：

def set_axes_height(ax, h):
    fig = ax.figure
    aw, ah = np.diff(ax.transAxes.transform([(0, 0), (1, 1)]), axis=0)[0]
    fw, fh = fig.get_size_inches()
    dpi = fig.get_dpi()
    scale = h / (ah / dpi)
    fig.set_size_inches(fw*scale, fh*scale, forward=True)