Gráfica de colores del array 2D Matplotlib

Suraj Joshi 30 enero 2023
  1. Plotar un array 2D en Matplotlib usando el método matplotlib.pyplot.imshow()
  2. Gráfico del array 2D en Matplotlib Usando el método matplotlib.pyplot.pcolormesh()
Gráfica de colores del array 2D Matplotlib

Este tutorial explica cómo podemos generar un gráfico de colores de matrices 2D utilizando los métodos matplotlib.pyplot.imshow() y matplotlib.pyplot.pcolormesh() en Python.

Plotar un array 2D en Matplotlib usando el método matplotlib.pyplot.imshow()

El método matplotlib.pyplot.imshow() toma un array 2D como entrada y renderiza el array dada como una imagen rasterizada.

La sintaxis de matplotlib.pyplot.imshow()

matplotlib.pyplot.imshow(X,
                         cmap=None,
                         norm=None,
                         aspect=None,
                         interpolation=None,
                         alpha=None,
                         vmin=None,
                         vmax=None,
                         origin=None,
                         extent=None, *,
                         filternorm=True,
                         filterrad=4.0,
                         resample=None,
                         url=None,
                         data=None,
                         **kwargs)

matplotlib.pyplot.imshow() Ejemplo

import numpy as np
import matplotlib.pyplot as plt

X = np.random.randint(256, size=(10, 10))

fig = plt.figure(figsize=(8, 6))
plt.imshow(X)
plt.title("Plot 2D array")
plt.show()

Producción:

Trazar el array 2D de Matplotlib usando imshow

Traza el array 2D creada usando el numpy.random.randint() de tamaño 10*10. Por defecto, los valores son mapeados usando el mapa de colores viridis.

Podemos establecer el parámetro cmap en el método imshow() para cambiar el mapa de colores.

import numpy as np
import matplotlib.pyplot as plt

X = np.random.randint(256, size=(10, 10))

fig = plt.figure(figsize=(8, 6))
plt.imshow(X, cmap="inferno")
plt.title("Plot 2D array")
plt.colorbar()
plt.show()

Producción:

Establecer cmap Plot 2D array Matplotlib usando imshow

Muestra el gráfico del array 2D con el mapa de colores inferno. También podemos ver una barra de color a la derecha del gráfico, que nos dice qué valores del array están mapeados a qué colores.

Gráfico del array 2D en Matplotlib Usando el método matplotlib.pyplot.pcolormesh()

La función matplotlib.pyplot.pcolormesh() crea una trama de pseudocolores en Matplotlib. Es similar a la función matplotlib.pyplot.pcolor().

import numpy as np
import matplotlib.pyplot as plt

X = np.random.randint(256, size=(10, 10))

fig = plt.figure(figsize=(8, 6))
plt.pcolormesh(X, cmap="plasma")
plt.title("Plot 2D array")
plt.colorbar()
plt.show()

Producción:

Trazar el array 2D de Matplotlib usando pcolormesh

Traza el array 2D creada usando el numpy.random.randint() de tamaño 10*10 con el mapa de color plasma. La barra de color de la derecha representa los colores asignados a diferentes rangos de valores.

Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn