Última actualización: 13 de diciembre de 2023

4.1. Matplotlib

Nota

Propósito: es una librería de Python especializada en la creación de gráficos en dos dimensiones.

Permite crear y personalizar los tipos de gráficos más comunes, entre ellos:

  • Diagramas de barras.

  • Histograma.

  • Diagramas de sectores.

  • Diagramas de caja y bigotes.

  • Diagramas de violín.

  • Diagramas de dispersión o puntos.

  • Diagramas de lineas.

  • Diagramas de areas.

  • Diagramas de contorno.

  • Mapas de color.

y combinaciones de todos ellos.

En la siguiente galería de gráficos pueden apreciarse todos los tipos de gráficos que pueden crearse con esta librería.

../_images/matplotlib_logo.png

Figura 4.1, Figura 4.1, Logotipo de librería Matplotlib

4.1.1. Instalación

Para instalar el paquete matplotlib ejecute el siguiente comando, el cual a continuación se presentan el correspondiente comando de tu sistema operativo:

$ pip install matplotlib

Puede probar si la instalación se realizo correctamente, ejecutando el siguiente comando correspondiente a tu sistema operativo:

$ python -c "import matplotlib ; print(matplotlib.__version__)"

Si muestra el numero de la versión instalada de matplotlib, tiene correctamente instalada la paquete. Con esto, ya tiene todo listo para continuar.

4.1.2. Creación de gráficos

Para crear un gráfico con matplotlib es habitual seguir los siguientes pasos:

  1. Importar el módulo pyplot.

  2. Definir la figura que contendrá el gráfico, que es

    la region (ventana o página) donde se dibujará y los ejes sobre los que se dibujarán los datos. Para ello se utiliza la función subplots().

  3. Dibujar los datos sobre los ejes. Para ello se

    utilizan distintas funciones dependiendo del tipo de gráfico que se quiera.

  4. Personalizar el gráfico. Para ello existen multitud

    de funciones que permiten añadir un título, una leyenda, una rejilla, cambiar colores o personalizar los ejes.

  5. Guardar el gráfico. Para ello se utiliza la función

    savefig().

  6. Mostrar el gráfico. Para ello se utiliza la función

    show().

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
# Importar el módulo pyplot con el alias plt
import matplotlib.pyplot as plt
# Crear la figura y los ejes
fig, ax = plt.subplots()
# Dibujar puntos
ax.scatter(x = [1, 2, 3], y = [3, 2, 1])
# Guardar el gráfico en formato png
plt.savefig('diagrama-dispersion.png')
# Mostrar el gráfico
plt.show()
../_images/matplotlib_diagrama_dispersion.png

Figura 4.2, Figura 4.2, Creación de gráficos

4.1.3. Diagramas de dispersión o puntos

  • scatter(x, y): Dibuja un diagrama de puntos con las coordenadas de la lista x en el eje X y las coordenadas de la lista y en el eje Y.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.scatter.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.scatter([1, 2, 3, 4], [1, 2, 0, 0.5])
plt.show()
../_images/matplotlib_diagrama_puntos.png

Figura 4.3, Figura 4.3, Diagramas de dispersión o puntos

4.1.4. Diagramas de líneas

  • plot(x, y): Dibuja un polígono con los vértices dados por las coordenadas de la lista x en el eje X y las coordenadas de la lista y en el eje Y.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.plot.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.plot([1, 2, 3, 4], [1, 2, 0, 0.5])
plt.show()
../_images/matplotlib_diagrama_lineas.png

Figura 4.4, Figura 4.4, Diagramas de líneas

4.1.5. Diagramas de areas

  • fill_between(x, y): Dibuja el area bajo el polígono con los vértices dados por las coordenadas de la lista x en el eje X y las coordenadas de la lista y en el eje Y.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.fill_between.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.fill_between([1, 2, 3, 4], [1, 2, 0, 0.5])
plt.show()
../_images/matplotlib_diagrama_lineas.png

Figura 4.5, Figura 4.5, Diagramas de areas

4.1.6. Diagramas de barras verticales

  • bar(x, y): Dibuja un diagrama de barras verticales donde x es una lista con la posición de las barras en el eje X, e y es una lista con la altura de las barras en el eje Y.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.bar.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.bar([1, 2, 3], [3, 2, 1])
plt.show()
../_images/matplotlib_diagrama_barras.png

Figura 4.6, Figura 4.6, Diagramas de barras verticales

4.1.7. Diagramas de barras horizontales

  • barh(x, y): Dibuja un diagrama de barras horizontales donde x es una lista con la posición de las barras en el eje Y, e y es una lista con la longitud de las barras en el eje X.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.barh.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.barh([1, 2, 3], [3, 2, 1])
plt.show()
../_images/matplotlib_diagrama_barras_horizontales.png

Figura 4.7, Figura 4.7, Diagramas de barras horizontales

4.1.8. Histogramas

  • hist(x, bins): Dibuja un histograma con las frecuencias resultantes de agrupar los datos de la lista x en las clases definidas por la lista bins.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.hist.

1
2
3
4
5
6
import numpy as np
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
x = np.random.normal(5, 1.5, size=1000)
ax.hist(x, np.arange(0, 11))
plt.show()
../_images/matplotlib_histograma.png

Figura 4.8, Figura 4.8, Histogramas

4.1.9. Diagramas de sectores

  • pie(x): Dibuja un diagrama de sectores con las frecuencias de la lista x.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.pie.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.pie([5, 4, 3, 2, 1])
plt.show()
../_images/matplotlib_diagrama_sectores.png

Figura 4.9, Figura 4.9, Diagramas de sectores

4.1.10. Diagramas de caja y bigotes

  • boxplot(x): Dibuja un diagrama de caja y bigotes con los datos de la lista x.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.boxplot.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.boxplot([1, 2, 1, 2, 3, 4, 3, 3, 5, 7])
plt.show()
../_images/matplotlib_diagrama_caja.png

Figura 4.10, Figura 4.10, Diagramas de caja y bigotes

4.1.11. Diagramas de violín

  • violinplot(x): Dibuja un diagrama de violín con los datos de la lista x.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.violinplot.

1
2
3
4
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.violinplot([1, 2, 1, 2, 3, 4, 3, 3, 5, 7])
plt.show()
../_images/matplotlib_diagrama_violin.png

Figura 4.11, Figura 4.11, Diagramas de violín

4.1.12. Diagramas de contorno

  • contourf(x, y, z): Dibuja un diagrama de contorno con las curvas de nivel de la superficie dada por los puntos con las coordenadas de las listas x, y y z en los ejes X, Y y Z respectivamente.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.contourf.

1
2
3
4
5
6
7
8
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
x = np.linspace(-3.0, 3.0, 100)
y = np.linspace(-3.0, 3.0, 100)
x, y = np.meshgrid(x, y)
z = np.sqrt(x**2 + 2*y**2)
ax.contourf(x, y, z)
plt.show()
../_images/matplotlib_diagrama_contorno.png

Figura 4.12, Figura 4.12, Diagramas de contorno

4.1.13. Mapas de color

  • imshow(x): Dibuja un mapa de color a partir de una matriz (array bidimensiona) x.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.imshow.

1
2
3
4
5
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
x = np.random.random((16, 16))
ax.imshow(x)
plt.show()
../_images/matplotlib_mapa_calor.png

Figura 4.13, Figura 4.13, Mapas de color

  • hist2d(x, y): Dibuja un mapa de color que simula un histograma bidimensional, donde los colores de los cuadrados dependen de las frecuencias de las clases de la muestra dada por las listas x e y.

    Truco

    Para más información consulte la documentación oficial de matplotlib.pyplot.hist2d.

1
2
3
4
5
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
x, y = np.random.multivariate_normal(mean=[0.0, 0.0], cov=[[1.0, 0.4], [0.4, 0.5]], size=1000).T
ax.hist2d(x, y)
plt.show()
../_images/matplotlib_histograma_2d.png

Figura 4.14, Figura 4.14, Mapas de color 2D

4.1.14. Cambiar el aspecto de los gráficos

Los gráficos creados con Matplotlib son personalizables y puede cambiarse el aspecto de casi todos sus elementos. Los elementos que suelen modificarse más a menudo son:

  • Colores

  • Marcadores de puntos

  • Estilo de líneas

  • Títulos

  • Ejes

  • Leyenda

  • Rejilla

4.1.15. Colores

Para cambiar el color de los objetos se utiliza el parámetro color = nombre-color, donde nombre-color es una cadena con el nombre del color de entre los colores disponibles.

1
2
3
4
5
6
7
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax.plot(dias, temperaturas['Madrid'], color = 'tab:purple')
ax.plot(dias, temperaturas['Barcelona'], color = 'tab:green')
plt.show()
../_images/matplotlib_diagrama_lineas_colores.png

Figura 4.15, Figura 4.15, Colores

4.1.16. Marcadores

Para cambiar la forma de los puntos marcadores se utiliza el parámetro marker = nombre-marcador donde nombre-marcador es una cadena con el nombre del marcador de entre los marcadores disponibles

1
2
3
4
5
6
7
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax.plot(dias, temperaturas['Madrid'], marker = '^')
ax.plot(dias, temperaturas['Barcelona'], marker = 'o')
plt.show()
../_images/matplotlib_lineas_marcadores.png

Figura 4.16, Figura 4.16, Marcadores

4.1.17. Líneas

Para cambiar el estilo de las líneas se utiliza el parámetro linestyle = nombre-estilo donde nombre-estilo es una cadena con el nombre del estilo de entre los estilos disponibles

1
2
3
4
5
6
7
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax.plot(dias, temperaturas['Madrid'], linestyle = 'dashed')
ax.plot(dias, temperaturas['Barcelona'], linestyle = 'dotted')
plt.show()
../_images/matplotlib_lineas_estilo.png

Figura 4.17, Figura 4.17, Líneas

4.1.18. Títulos

Para añadir un título principal al gráfico se utiliza el siguiente método:

  • ax.set_title(titulo, loc=alineacion, fontdict=fuente)

    : Añade un título con el contenido de la cadena titulo a los ejes ax. El parámetro loc indica la alineación del título, que puede ser 'left' (izquierda), 'center' (centro) o 'right' (derecha), y el parámetro fontdict indica mediante un diccionario las características de la fuente (la el tamaño fontisize, el grosor fontweight o el color color).

1
2
3
4
5
6
7
8
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax.plot(dias, temperaturas['Madrid'])
ax.plot(dias, temperaturas['Barcelona'])
ax.set_title('Evolución de la temperatura diaria', loc = "left", fontdict = {'fontsize':14, 'fontweight':'bold', 'color':'tab:blue'})
plt.show()
../_images/matplotlib_titulo.png

Figura 4.18, Figura 4.18, Títulos

4.1.19. Ejes

Para cambiar el aspecto de los ejes se suelen utilizar los siguientes métodos:

  • ax.set_xlabel(titulo): Añade un título con el contenido de la cadena titulo al eje x de ax. Se puede personalizar la alineación y la fuente con los mismos parámetros que para el título principal.

  • ax.set_ylabel(titulo): Añade un título con el contenido de la cadena titulo al eje y de ax. Se puede personalizar la alineación y la fuente con los mismos parámetros que para el título principal.

  • ax.set_xlim([limite-inferior, limite-superior]): Establece los límites que se muestran en el eje x de ax.

  • ax.set_ylim([limite-inferior, limite-superior]): Establece los límites que se muestran en el eje y de ax.

  • ax.set_xticks(marcas): Dibuja marcas en el eje x de ax en las posiciones indicadas en la lista marcas.

  • ax.set_yticks(marcas): Dibuja marcas en el eje y de ax en las posiciones indicadas en la lista marcas.

  • ax.set_xscale(escala): Establece la escala del eje x de ax, donde el parámetro escala puede ser 'linear' (lineal) o 'log' (logarítmica).

  • ax.set_yscale(escala): Establece la escala del eje y de ax, donde el parámetro escala puede ser 'linear' (lineal) o 'log' (logarítmica).

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax.plot(dias, temperaturas['Madrid'])
ax.plot(dias, temperaturas['Barcelona'])
ax.set_xlabel("Días", fontdict = {'fontsize':14, 'fontweight':'bold', 'color':'tab:blue'})
ax.set_ylabel("Temperatura ºC")
ax.set_ylim([20,35])
ax.set_yticks(range(20, 35))
plt.show()
../_images/matplotlib_ejes.png

Figura 4.19, Figura 4.19, Ejes

4.1.20. Leyenda

Para añadir una leyenda a un gráfico se utiliza el siguiente método:

  • ax.legend(leyendas, loc = posición): Dibuja un leyenda en los ejes ax con los nombres indicados en la lista leyendas. El parámetro loc indica la posición en la que se dibuja la leyenda y puede ser 'upper left' (arriba izquierda), 'upper center' (arriba centro), 'upper right' (arriba derecha), 'center left' (centro izquierda), 'center' (centro), 'center right' (centro derecha), 'lower left' (abajo izquierda), 'lower center' (abajo centro), 'lower right' (abajo derecha). Se puede omitir la lista leyendas si se indica la leyenda de cada serie en la función que la dibuja mediante el parámetro label.

1
2
3
4
5
6
7
8
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax.plot(dias, temperaturas['Madrid'], label = 'Madrid')
ax.plot(dias, temperaturas['Barcelona'], label = 'Barcelona')
ax.legend(loc = 'upper right')
plt.show()
../_images/matplotlib_leyenda.png

Figura 4.20, Figura 4.20, Leyenda

4.1.21. Rejilla

ax.grid(axis=ejes, color=color, linestyle=estilo): Dibuja una rejilla en los ejes de ax. El parámetro axis indica los ejes sobre los que se dibuja la regilla y puede ser 'x' (eje x), 'y' (eje y) o 'both' (ambos). Los parámetros color y linestyle establecen el color y el estilo de las líneas de la rejilla, y pueden tomar los mismos valores vistos en los apartados de colores y líneas.

1
2
3
4
5
6
7
8
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax.plot(dias, temperaturas['Madrid'])
ax.plot(dias, temperaturas['Barcelona'])
ax.grid(axis = 'y', color = 'gray', linestyle = 'dashed')
plt.show()
../_images/matplotlib_rejilla.png

Figura 4.21, Figura 4.21, Rejilla

4.1.22. Múltiples gráficos

Es posible dibujar varios gráficos en distintos ejes en una misma figura organizados en forma de tabla. Para ello, cuando se inicializa la figura y los ejes, hay que pasarle a la función subplots el número de filas y columnas de la tabla que contendrá los gráficos. Con esto los distintos ejes se organizan en un array y se puede acceder a cada uno de ellos a través de sus índices. Si se quiere que los distintos ejes compartan los mismos límites para los ejes se pueden pasar los parámetros sharex = True para el eje x o sharey = True para el eje y.

1
2
3
4
5
6
7
8
9
import matplotlib.pyplot as plt
fig, ax = plt.subplots(2, 2, sharey = True)
dias = ['L', 'M', 'X', 'J', 'V', 'S', 'D']
temperaturas = {'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5], 'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]}
ax[0, 0].plot(dias, temperaturas['Madrid'])
ax[0, 1].plot(dias, temperaturas['Barcelona'], color = 'tab:orange')
ax[1, 0].bar(dias, temperaturas['Madrid'])
ax[1, 1].bar(dias, temperaturas['Barcelona'], color = 'tab:orange')
plt.show()
../_images/matplotlib_multiples_graficos.png

Figura 4.22, Figura 4.22, Múltiples gráficos

4.1.23. Integración con Pandas

Matplotlib se integra a la perfección con la librería Pandas, permitiendo dibujar gráficos a partir de los datos de las series y DataFrames de Pandas.

  • df.plot(kind=tipo, x=columnax, y=columnay, ax=ejes): Dibuja un diagrama del tipo indicado por el parámetro kind en los ejes indicados en el parámetro ax, representando en el eje x la columna del parámetro x y en el eje y la columna del parámetro y. El parámetro kind puede tomar como argumentos 'line' (lineas), 'scatter' (puntos), 'bar' (barras verticales), 'barh' (barras horizontales), 'hist' (histograma), 'box' (cajas), 'density' (densidad), 'area' (area) o 'pie' (sectores). Es posible pasar otros parámetros para indicar el color, el marcador o el estilo de línea como se vió en los apartados anteriores.

1
2
3
4
5
6
7
8
9
import pandas as pd
import matplotlib.pyplot as plt
df = pd.DataFrame({'Días':['L', 'M', 'X', 'J', 'V', 'S', 'D'],
                'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5],
                'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]})
fig, ax = plt.subplots()
df.plot(x = 'Días', y = 'Madrid', ax = ax)
df.plot(x = 'Días', y = 'Barcelona', ax = ax)
plt.show()
../_images/matplotlib_integrar_pandas.png

Figura 4.23, Figura 4.23, Integración con Pandas

Si no se indican los parámetros x e y se representa el índice de las filas en el eje x y una serie por cada columna del Dataframe. Las columnas no numéricas se ignoran.

1
2
3
4
5
6
7
8
9
import pandas as pd
import matplotlib.pyplot as plt
df = pd.DataFrame({'Días':['L', 'M', 'X', 'J', 'V', 'S', 'D'],
                'Madrid':[28.5, 30.5, 31, 30, 28, 27.5, 30.5],
                'Barcelona':[24.5, 25.5, 26.5, 25, 26.5, 24.5, 25]})
df = df.set_index('Días')
fig, ax = plt.subplots()
df.plot(ax = ax)
plt.show()
../_images/matplotlib_integrar_pandas2.png

Figura 4.24, Figura 4.24, Integración con Pandas

Por hacer

TODO Terminar de escribir esta sección.


Ver también

Consulte la sección de lecturas suplementarias del entrenamiento para ampliar su conocimiento en esta temática.


¿Cómo puedo ayudar?

¡Mi soporte está aquí para ayudar!

Mi horario de oficina es de lunes a sábado, de 9 AM a 5 PM. GMT-4 - Caracas, Venezuela.

La hora aquí es actualmente 7:35 PM GMT-4.

Mi objetivo es responder a todos los mensajes dentro de un día hábil.

Contrata mi increíble soporte profesional