plt.subplot_adjust () não está funcionando corretamente

Estou fazendo alguns gráficos de densidade assim:

import matplotlib.pyplot as plt
import numpy as np
from sklearn.metrics import r2_score
import matplotlib
from scipy import stats
import matplotlib.gridspec as gridspec
from mpl_toolkits.axes_grid1.inset_locator import InsetPosition
from matplotlib.ticker import FormatStrFormatter
import matplotlib.cm as cm
from scipy.ndimage.filters import gaussian_filter
import random

matplotlib.rcParams.update({'font.size': 16})
matplotlib.rcParams['xtick.direction'] = 'in'
matplotlib.rcParams['ytick.direction'] = 'in'

x = random.sample(range(1, 10001), 1000)
y = random.sample(range(1, 10001), 1000)



def myplot(x, y, s, bins=1000):
    heatmap, xedges, yedges = np.histogram2d(x, y, bins=bins)
    heatmap = gaussian_filter(heatmap, sigma=s)

    extent = [xedges[0], xedges[-1], yedges[0], yedges[-1]]
    return heatmap.T, extent
cmap = cm.YlOrRd

fig, (ax, ax1, cax) = plt.subplots(ncols = 3, figsize = (15, 5),
                       gridspec_kw={"width_ratios":[1,1, 0.5]})
img, extent = myplot(x, y, 20)
im = ax.imshow(img, extent = extent, origin = 'lower', cmap = cmap)
ax.text(0.05, 0.92, '$R^2$ = {}'.format(np.round(r2_score(x, y), 2)), fontsize=14, color = 'k', transform = ax.transAxes)
ax.plot(ax.get_xlim(), ax.get_ylim(), ls="--", c=".3")
ax.set_xlabel("Black Sky")
ax.set_ylabel("Blue Sky")

img2, extent2 = myplot(x, y, 20)
ax1.imshow(img2, extent = extent2, origin = 'lower', cmap = cmap)
ax1.text(0.05, 0.92, '$R^2$ = {}'.format(np.round(r2_score(x, y), 2)), fontsize=14, color = 'k', transform = ax1.transAxes)
ax1.axes.get_yaxis().set_visible(False)
ax1.yaxis.set_ticks([])
ax1.plot(ax1.get_xlim(), ax1.get_ylim(), ls="--", c=".3")
ax1.set_xlabel("White Sky")


ip = InsetPosition(ax1, [1.05,0,0.05,1]) 

cax.set_axes_locator(ip)

fig.colorbar(im, cax=cax, ax=[ax,ax1], use_gridspec = True)

plt.subplots_adjust(wspace=0.1, hspace=0)

o que me dá um enredo como este:

Não importa o que eu mudewspace para o enredo permanece o mesmo. Eu acho que é porque quando eu viro o eixo y emax1 Estou apenas deixando o texto em branco em vez de remover o eixo y todos juntos. Existe uma maneira de fazer isso para que eu possa aproximar o espaçamento entre as figuras?

questionAnswers(1)

yourAnswerToTheQuestion