Creación de subclases de figura personalizadas

Introducción

En este laboratorio, aprenderá a crear subclases de figura personalizadas en Matplotlib. Creará una clase WatermarkFigure que agrega una marca de agua de texto a la trama.

Consejos sobre la VM

Una vez finalizada la inicialización de la VM, haga clic en la esquina superior izquierda para cambiar a la pestaña Cuaderno y acceder a Jupyter Notebook para practicar.

A veces, es posible que tenga que esperar unos segundos a que Jupyter Notebook termine de cargarse. La validación de operaciones no puede automatizarse debido a las limitaciones de Jupyter Notebook.

Si tiene problemas durante el aprendizaje, no dude en preguntar a Labby. Deje comentarios después de la sesión y lo resolveremos rápidamente para usted.

Importar las bibliotecas necesarias

Primero, importe las bibliotecas necesarias: matplotlib.pyplot y numpy.

import matplotlib.pyplot as plt
import numpy as np

Crear una subclase de figura personalizada

Cree una subclase de figura personalizada llamada WatermarkFigure que agregue una marca de agua de texto a la trama. Esta clase hereda de la clase Figure de Matplotlib.

from matplotlib.figure import Figure

class WatermarkFigure(Figure):
    """Una figura con una marca de agua de texto."""

    def __init__(self, *args, watermark=None, **kwargs):
        super().__init__(*args, **kwargs)

        if watermark is not None:
            bbox = dict(boxstyle='square', lw=3, ec='gray',
                        fc=(0.9, 0.9,.9,.5), alpha=0.5)
            self.text(0.5, 0.5, watermark,
                      ha='center', va='center', rotation=30,
                      fontsize=40, color='gray', alpha=0.5, bbox=bbox)

Crear datos para la trama

Cree algunos datos para la trama. En este ejemplo, crearemos los arrays x e y utilizando la biblioteca numpy.

x = np.linspace(-3, 3, 201)
y = np.tanh(x) + 0.1 * np.cos(5 * x)

Graficar los datos utilizando la subclase de figura personalizada

Utilice la función plt.figure() para graficar los datos utilizando la subclase de figura personalizada WatermarkFigure. En este ejemplo, agregaremos el texto de la marca de agua "borrador" a la trama.

plt.figure(FigureClass=WatermarkFigure, watermark='draft')
plt.plot(x, y)

Revisar referencias

Revise las referencias utilizadas en este ejemplo.

## Referencias
## matplotlib.pyplot.figure
## matplotlib.figure.Figure
## matplotlib.figure.Figure.text

Resumen

En este laboratorio, aprendiste cómo crear una subclase de figura personalizada en Matplotlib. Creaste una clase WatermarkFigure que agrega una marca de agua de texto a la trama. También aprendiste cómo graficar datos utilizando la subclase de figura personalizada.