def set_ax_size(ax, w, h):
    """Width/Height in display (pixel) coords."""
    fig = ax.figure
    w, h = fig.transFigure.inverted().transform([w, h])
    x, y, _, _ = ax.get_position().bounds
    ax.set_position([x, y, w, h])
