[PYTHON] Définissez les paramètres communs avec le sous-graphique de matplotlib

Chose que tu veux faire

Je souhaite définir des paramètres communs tels que des grilles et des étiquettes pour tous les axes dessinés avec le sous-tracé de matplotlib.

Méthode 1

1. Obtenez une liste d'axes.
2. Boucle pour chaque axe
3. Changer l'axe actuel

# 1.Obtenez une liste d'axes
axs = plt.gcf().get_axes()

# 2.Boucle pour chaque axe
for ax in axs:
    # 3.Changer l'axe actuel
    plt.axes(ax)

Exemple de code

import matplotlib.pyplot as plt
import numpy as np

#Les données
x  = np.linspace(-np.pi, np.pi, 100)
y1 = np.sin(x)
y2 = np.cos(x)

# figure
plt.figure(figsize=(8,4))

#Parcelle 1
plt.subplot(1,2,1)
plt.plot(x,y1,label='sin')

#Parcelle 2
plt.subplot(1,2,2)
plt.plot(x,y2,label='cos')

#Obtenez une liste d'axes
axs = plt.gcf().get_axes()

#Boucle pour chaque axe
for ax in axs:
    #Changer l'axe actuel
    plt.axes(ax)

    #Afficher la légende
    plt.legend(loc=2)

    #la grille
    plt.grid(linestyle='--')

    #Étiquette d'axe
    plt.xlabel('x')
    plt.ylabel('y')

    #Gamme d'axes
    plt.xlim([-np.pi, np.pi])
    plt.ylim([  -1.2,   1.2])

#Ajustement de la figure
plt.tight_layout()

Méthode 2

1. Répertoriez les données et les étiquettes.
2. Boucle avec enumerate
3. N'oubliez pas d'ajouter +1 à plt.subplot

# 1.Répertoriez les données et les étiquettes.
Y    = [y1, y2]
lbls = ['sin', 'cos']

# 2.Boucle en utilisant enumerate
for i, y in enumerate(Y):

    # 3. plt.subplot est+N'oubliez pas de
    plt.subplot(1,2,i+1)

Exemple de code 2

# figure
plt.figure(figsize=(8,4))
# list
Y    = [y1, y2]
lbls = ['sin', 'cos']

for i, y in enumerate(Y):
    #Sous-parcelle
    plt.subplot(1,2,i+1)
    
    #terrain
    plt.plot(x,y,label=lbls[i])

    #Afficher la légende
    plt.legend(loc=2)

    #la grille
    plt.grid(linestyle='--')

    #Étiquette d'axe
    plt.xlabel('x')
    plt.ylabel('y')

    #Gamme d'axes
    plt.xlim([-np.pi, np.pi])
    plt.ylim([  -1.2,   1.2])

#Ajustement de la figure
plt.tight_layout()

Les informations complémentaires suivantes

gca, gcf Vous pouvez obtenir la figure courante avec plt.gcf () et les axes courants avec plt.gca (). Vous pouvez également spécifier l'axe actuel avec plt.axes ().

Structure hiérarchique

La figure de matplotlib a une structure hiérarchique comme le montre la figure ci-dessous, comme MATLAB.

Il y a des axes sur la figure et des lignes dans les axes. Vous pouvez obtenir une liste de tous les axes de la figure courante avec plt.gcf (). Get_axes (). De plus, plt.gca (). Get_lines () peut récupérer toutes les lignes dans les axes courants.