Xlabel / ylabel commun pour les sous-graphiques matplotlib

J'ai l'intrigue suivante:

fig,ax = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

et maintenant je voudrais donner à ce tracé des étiquettes communes sur l'axe des x et des étiquettes sur l'axe des y. Avec "commun", je veux dire qu'il devrait y avoir une grande étiquette de l'axe des x sous toute la grille des sous-graphiques, et une grande étiquette de l'axe des y vers la droite. Je ne trouve rien à ce sujet dans la documentation de plt.subplots, et mes googlings suggèrent que je dois faire un gros plt.subplot(111)pour commencer - mais comment puis-je ensuite mettre mes sous-parcelles 5 * 2 dans cela en utilisant plt.subplots?

Cela ressemble à ce que vous voulez réellement. Il applique la même approche de cette réponse à votre cas spécifique:

import matplotlib.pyplot as plt

fig, ax = plt.subplots(nrows=3, ncols=3, sharex=True, sharey=True, figsize=(6, 6))

fig.text(0.5, 0.04, 'common X', ha='center')
fig.text(0.04, 0.5, 'common Y', va='center', rotation='vertical')

Comme je le considère suffisamment pertinent et élégant (pas besoin de spécifier les coordonnées pour placer le texte), je copie (avec une légère adaptation) une réponse à une autre question connexe .

import matplotlib.pyplot as plt
fig, axes = plt.subplots(5, 2, sharex=True, sharey=True, figsize=(6,15))
# add a big axis, hide frame
fig.add_subplot(111, frameon=False)
# hide tick and tick label of the big axis
plt.tick_params(labelcolor='none', top=False, bottom=False, left=False, right=False)
plt.xlabel("common X")
plt.ylabel("common Y")

Il en résulte ce qui suit (avec matplotlib version 2.2.0):

Sans sharex=True, sharey=Truevous obtenez:

Avec lui, vous devriez le rendre plus agréable:

fig, axes2d = plt.subplots(nrows=3, ncols=3,
                           sharex=True, sharey=True,
                           figsize=(6,6))

for i, row in enumerate(axes2d):
    for j, cell in enumerate(row):
        cell.imshow(np.random.rand(32,32))

plt.tight_layout()

Mais si vous souhaitez ajouter des étiquettes supplémentaires, vous ne devez les ajouter qu'aux tracés d'arête:

fig, axes2d = plt.subplots(nrows=3, ncols=3,
                           sharex=True, sharey=True,
                           figsize=(6,6))

for i, row in enumerate(axes2d):
    for j, cell in enumerate(row):
        cell.imshow(np.random.rand(32,32))
        if i == len(axes2d) - 1:
            cell.set_xlabel("noise column: {0:d}".format(j + 1))
        if j == 0:
            cell.set_ylabel("noise row: {0:d}".format(i + 1))

plt.tight_layout()

Ajouter une étiquette pour chaque tracé le gâcherait (il existe peut-être un moyen de détecter automatiquement les étiquettes répétées, mais je n'en ai pas connaissance).

Depuis la commande:

fig,ax = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

vous avez utilisé renvoie un tuple composé de la figure et d'une liste des instances d'axes, il suffit déjà de faire quelque chose comme (attention, j'ai changé fig,axen fig,axes):

fig,axes = plt.subplots(5,2,sharex=True,sharey=True,figsize=fig_size)

for ax in axes:
    ax.set_xlabel('Common x-label')
    ax.set_ylabel('Common y-label')

Si vous souhaitez modifier certains détails sur une sous-parcelle spécifique, vous pouvez y accéder via axes[i]où iitère sur vos sous-parcelles.

Il peut également être très utile d'inclure un

fig.tight_layout()

à la fin du fichier, avant le plt.show(), afin d'éviter le chevauchement des étiquettes.

Cela sera mieux si vous réservez de l'espace pour les étiquettes communes en créant des étiquettes invisibles pour le sous-tracé dans le coin inférieur gauche. Il est également bon de passer la taille de la police depuis rcParams. De cette façon, les étiquettes communes changeront de taille avec votre configuration rc, et les axes seront également ajustés pour laisser de l'espace pour les étiquettes communes.

fig_size = [8, 6]
fig, ax = plt.subplots(5, 2, sharex=True, sharey=True, figsize=fig_size)
# Reserve space for axis labels
ax[-1, 0].set_xlabel('.', color=(0, 0, 0, 0))
ax[-1, 0].set_ylabel('.', color=(0, 0, 0, 0))
# Make common axis labels
fig.text(0.5, 0.04, 'common X', va='center', ha='center', fontsize=rcParams['axes.labelsize'])
fig.text(0.04, 0.5, 'common Y', va='center', ha='center', rotation='vertical', fontsize=rcParams['axes.labelsize'])

J'ai rencontré un problème similaire en traçant une grille de graphiques. Les graphiques se composaient de deux parties (haut et bas). L'étiquette y était censée être centrée sur les deux parties.

Je ne voulais pas utiliser une solution qui dépend de la connaissance de la position dans la figure extérieure (comme fig.text ()), j'ai donc manipulé la position y de la fonction set_ylabel (). Il est généralement égal à 0,5, au milieu du graphique auquel il est ajouté. Comme le remplissage entre les parties (hspace) dans mon code était nul, j'ai pu calculer le milieu des deux parties par rapport à la partie supérieure.

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

# Create outer and inner grid
outerGrid = gridspec.GridSpec(2, 3, width_ratios=[1,1,1], height_ratios=[1,1])
somePlot = gridspec.GridSpecFromSubplotSpec(2, 1,
               subplot_spec=outerGrid[3], height_ratios=[1,3], hspace = 0)

# Add two partial plots
partA = plt.subplot(somePlot[0])
partB = plt.subplot(somePlot[1])

# No x-ticks for the upper plot
plt.setp(partA.get_xticklabels(), visible=False)

# The center is (height(top)-height(bottom))/(2*height(top))
# Simplified to 0.5 - height(bottom)/(2*height(top))
mid = 0.5-somePlot.get_height_ratios()[1]/(2.*somePlot.get_height_ratios()[0])
# Place the y-label
partA.set_ylabel('shared label', y = mid)

plt.show()

image

Inconvénients:

• La distance horizontale par rapport au tracé est basée sur la partie supérieure, les graduations inférieures peuvent s'étendre dans l'étiquette.

• La formule ne tient pas compte de l'espace entre les pièces.

• Lève une exception lorsque la hauteur de la partie supérieure est de 0.

Il existe probablement une solution générale qui prend en compte le remplissage entre les chiffres.

Mettre à jour:

Cette fonctionnalité fait maintenant partie du paquet proplot matplotlib que j'ai récemment publié sur pypi. Par défaut, lorsque vous créez des figures, les étiquettes sont "partagées" entre les axes.

Réponse originale:

J'ai découvert une méthode plus robuste:

Si vous connaissez les kwargs bottomet topqui sont entrés dans une GridSpecinitialisation, ou si vous connaissez les positions des bords de vos axes en Figurecoordonnées , vous pouvez également spécifier la position de l'étiquette y en Figurecoordonnées avec une magie de «transformation» sophistiquée. Par exemple:

import matplotlib.transforms as mtransforms
bottom, top = .1, .9
f, a = plt.subplots(nrows=2, ncols=1, bottom=bottom, top=top)
avepos = (bottom+top)/2
a[0].yaxis.label.set_transform(mtransforms.blended_transform_factory(
       mtransforms.IdentityTransform(), f.transFigure # specify x, y transform
       )) # changed from default blend (IdentityTransform(), a[0].transAxes)
a[0].yaxis.label.set_position((0, avepos))
a[0].set_ylabel('Hello, world!')

... et vous devriez voir que l'étiquette s'ajuste toujours de manière appropriée de gauche à droite pour éviter de se chevaucher avec les étiquettes, comme d'habitude - mais maintenant elle s'ajustera pour être toujours exactement entre les sous-graphiques souhaités.

De plus, si vous ne l'utilisez même pas set_position, le ylabel apparaîtra par défaut exactement à mi-hauteur de la figure . Je suppose que c'est parce que lorsque l'étiquette est finalement dessinée, matplotlibutilise 0,5 pour la y-coordonnée sans vérifier si la transformation de coordonnées sous-jacente a changé.