Gráfica de colores del array 2D Matplotlib

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()

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:

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:

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:

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.