Effet de zoom sur les axes

Introduction

Ce tutoriel vous guidera sur la manière de créer un effet de zoom à l'aide de matplotlib. L'effet de zoom vous permettra de vous connecter et de zoomer sur deux axes différents.

Conseils sur la machine virtuelle

Une fois le démarrage de la machine virtuelle terminé, cliquez sur le coin supérieur gauche pour basculer vers l'onglet Carnet de notes pour accéder à Jupyter Notebook pour pratiquer.

Parfois, vous devrez peut-être attendre quelques secondes pour que Jupyter Notebook ait fini de charger. La validation des opérations ne peut pas être automatisée en raison des limitations de Jupyter Notebook.

Si vous rencontrez des problèmes pendant l'apprentissage, n'hésitez pas à demander à Labby. Donnez votre feedback après la session, et nous résoudrons rapidement le problème pour vous.

Importation des bibliothèques requises

Dans cette étape, nous allons importer les bibliothèques nécessaires pour ce tutoriel. Nous allons importer matplotlib, et les bibliothèques pertinentes de mpl_toolkits.axes_grid1.

import matplotlib.pyplot as plt

from matplotlib.transforms import (Bbox, TransformedBbox,
                                   blended_transform_factory)
from mpl_toolkits.axes_grid1.inset_locator import (BboxConnector,
                                                   BboxConnectorPatch,
                                                   BboxPatch)

Définition de la connexion entre les axes

Dans cette étape, nous allons définir la connexion entre les deux axes. Cette fonction prend deux axes en entrée, ainsi que les valeurs minimales et maximales pour l'axe x. Elle crée ensuite une boîte englobante et connecte les deux axes.

def zoom_effect01(ax1, ax2, xmin, xmax, **kwargs):
    bbox = Bbox.from_extents(xmin, 0, xmax, 1)

    mybbox1 = TransformedBbox(bbox, ax1.get_xaxis_transform())
    mybbox2 = TransformedBbox(bbox, ax2.get_xaxis_transform())

    prop_patches = {**kwargs, "ec": "none", "alpha": 0.2}

    c1, c2, bbox_patch1, bbox_patch2, p = connect_bbox(
        mybbox1, mybbox2,
        loc1a=3, loc2a=2, loc1b=4, loc2b=1,
        prop_lines=kwargs, prop_patches=prop_patches)

    ax1.add_patch(bbox_patch1)
    ax2.add_patch(bbox_patch2)
    ax2.add_patch(c1)
    ax2.add_patch(c2)
    ax2.add_patch(p)

    return c1, c2, bbox_patch1, bbox_patch2, p

Création de la deuxième axe agrandi

Dans cette étape, nous allons créer la deuxième axe agrandi. Cette fonction prend deux axes en entrée. Elle crée ensuite une boîte englobante pour la deuxième axe et la connecte à la première axe.

def zoom_effect02(ax1, ax2, **kwargs):
    tt = ax1.transScale + (ax1.transLimits + ax2.transAxes)
    trans = blended_transform_factory(ax2.transData, tt)

    mybbox1 = ax1.bbox
    mybbox2 = TransformedBbox(ax1.viewLim, trans)

    prop_patches = {**kwargs, "ec": "none", "alpha": 0.2}

    c1, c2, bbox_patch1, bbox_patch2, p = connect_bbox(
        mybbox1, mybbox2,
        loc1a=3, loc2a=2, loc1b=4, loc2b=1,
        prop_lines=kwargs, prop_patches=prop_patches)

    ax1.add_patch(bbox_patch1)
    ax2.add_patch(bbox_patch2)
    ax2.add_patch(c1)
    ax2.add_patch(c2)
    ax2.add_patch(p)

    return c1, c2, bbox_patch1, bbox_patch2, p

Connexion des axes

Dans cette étape, nous allons connecter les axes et créer l'effet de zoom. Nous allons créer une figure avec quatre axes et les connecter en utilisant les fonctions zoom_effect01 et zoom_effect02.

axs = plt.figure().subplot_mosaic([
    ["zoom1", "zoom2"],
    ["main", "main"],
])

axs["main"].set(xlim=(0, 5))
zoom_effect01(axs["zoom1"], axs["main"], 0.2, 0.8)
axs["zoom2"].set(xlim=(2, 3))
zoom_effect02(axs["zoom2"], axs["main"])

plt.show()

Sommaire

Dans ce tutoriel, nous avons appris à créer un effet de zoom à l'aide de matplotlib. Nous avons d'abord défini une fonction pour connecter les axes, puis créé une deuxième fonction pour créer la deuxième axe agrandi. Enfin, nous avons connecté les axes et créé l'effet de zoom.