Comment supprimer les espaces entre les sous-graphiques dans matplotlib?

Le code ci-dessous produit des espaces entre les sous-graphiques. Comment supprimer les espaces entre les sous-graphiques et créer une grille serrée de l'image?

import matplotlib.pyplot as plt

for i in range(16):
    i = i + 1
    ax1 = plt.subplot(4, 4, i)
    plt.axis('on')
    ax1.set_xticklabels([])
    ax1.set_yticklabels([])
    ax1.set_aspect('equal')
    plt.subplots_adjust(wspace=None, hspace=None)
plt.show()

Vous pouvez utiliser gridspec pour contrôler l'espacement entre les axes. Il y a plus d' informations ici.

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

plt.figure(figsize = (4,4))
gs1 = gridspec.GridSpec(4, 4)
gs1.update(wspace=0.025, hspace=0.05) # set the spacing between axes. 

for i in range(16):
   # i = i + 1 # grid spec indexes from 0
    ax1 = plt.subplot(gs1[i])
    plt.axis('on')
    ax1.set_xticklabels([])
    ax1.set_yticklabels([])
    ax1.set_aspect('equal')

plt.show()

Le problème est l'utilisation de aspect='equal', qui empêche les sous-graphiques de s'étirer à un rapport hauteur / largeur arbitraire et de remplir tout l'espace vide.

Normalement, cela fonctionnerait:

import matplotlib.pyplot as plt

ax = [plt.subplot(2,2,i+1) for i in range(4)]

for a in ax:
    a.set_xticklabels([])
    a.set_yticklabels([])

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

Le résultat est le suivant:

Cependant, avec aspect='equal', comme dans le code suivant:

import matplotlib.pyplot as plt

ax = [plt.subplot(2,2,i+1) for i in range(4)]

for a in ax:
    a.set_xticklabels([])
    a.set_yticklabels([])
    a.set_aspect('equal')

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

Voici ce que nous obtenons:

La différence dans ce deuxième cas est que vous avez forcé les axes x et y à avoir le même nombre d'unités / pixel. Étant donné que les axes vont de 0 à 1 par défaut (c'est-à-dire avant de tracer quoi que ce soit), l'utilisation aspect='equal'force chaque axe à être un carré. Puisque la figure n'est pas un carré, pyplot ajoute un espacement supplémentaire entre les axes horizontalement.

Pour contourner ce problème, vous pouvez définir votre silhouette pour qu'elle ait le bon rapport hauteur / largeur. Nous allons utiliser ici l'interface pyplot orientée objet, que je considère comme supérieure en général:

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(8,8)) # Notice the equal aspect ratio
ax = [fig.add_subplot(2,2,i+1) for i in range(4)]

for a in ax:
    a.set_xticklabels([])
    a.set_yticklabels([])
    a.set_aspect('equal')

fig.subplots_adjust(wspace=0, hspace=0)

Voici le résultat:

Sans recourir entièrement à gridspec , les éléments suivants peuvent également être utilisés pour supprimer les espaces en définissant wspace et hspace sur zéro:

import matplotlib.pyplot as plt

plt.clf()
f, axarr = plt.subplots(4, 4, gridspec_kw = {'wspace':0, 'hspace':0})

for i, ax in enumerate(f.axes):
    ax.grid('on', linestyle='--')
    ax.set_xticklabels([])
    ax.set_yticklabels([])

plt.show()
plt.close()

Résultant en:

Avez-vous essayé plt.tight_layout()?

avec plt.tight_layout() sans elle:

Ou: quelque chose comme ça (utiliser add_axes)

left=[0.1,0.3,0.5,0.7]
width=[0.2,0.2, 0.2, 0.2]
rectLS=[]
for x in left:
   for y in left:
       rectLS.append([x, y, 0.2, 0.2])
axLS=[]
fig=plt.figure()
axLS.append(fig.add_axes(rectLS[0]))
for i in [1,2,3]:
     axLS.append(fig.add_axes(rectLS[i],sharey=axLS[-1]))    
axLS.append(fig.add_axes(rectLS[4]))
for i in [1,2,3]:
     axLS.append(fig.add_axes(rectLS[i+4],sharex=axLS[i],sharey=axLS[-1]))
axLS.append(fig.add_axes(rectLS[8]))
for i in [5,6,7]:
     axLS.append(fig.add_axes(rectLS[i+4],sharex=axLS[i],sharey=axLS[-1]))     
axLS.append(fig.add_axes(rectLS[12]))
for i in [9,10,11]:
     axLS.append(fig.add_axes(rectLS[i+4],sharex=axLS[i],sharey=axLS[-1]))

Si vous n'avez pas besoin de partager des axes, alors simplement axLS=map(fig.add_axes, rectLS)

Avec les versions récentes de matplotlib, vous voudrez peut-être essayer la mise en page contrainte . Cela ne fonctionne pas avec plt.subplot()cependant, vous devez donc utiliser à la plt.subplots()place:

fig, axs = plt.subplots(4, 4, constrained_layout=True)