Colorplot de 2D Array Matplotlib

1. Plotar matriz 2D em Matplotlib usando o método matplotlib.pyplot.imshow()
2. Lote 2D Array em Matplotlib Utilizando o matplotlib.pyplot.pcolormesh() Método

Este tutorial explica como podemos gerar colorplot plot de arrays 2D utilizando os métodos matplotlib.pyplot.imshow() e matplotlib.pyplot.pcolormesh() em Python.

Plotar matriz 2D em Matplotlib usando o método matplotlib.pyplot.imshow()

O método matplotlib.pyplot.imshow() toma um array 2D como entrada e torna o array dado como uma imagem raster.

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

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

Resultado:

Traça a matriz 2D criada utilizando o numpy.random.randint() de tamanho 10*10. Por defeito, os valores são mapeados utilizando o mapa de cores viridis.

Podemos definir o parâmetro cmap no método imshow() para alterar o mapa de cores.

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

Resultado:

Exibe o gráfico do array 2D com o mapa de cores inferno. Podemos também ver uma barra de cores no lado direito da parcela, que nos diz quais os valores no array são mapeados a que cores.

Lote 2D Array em Matplotlib Utilizando o matplotlib.pyplot.pcolormesh() Método

A função matplotlib.pyplot.pcolormesh() cria um gráfico pseudo-colorido em Matplotlib. É semelhante à função 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()

Resultado:

Traça a matriz 2D criada utilizando o numpy.random.randint() de tamanho 10*10 com o plasma colormap. A barra de cores à direita representa as cores atribuídas a diferentes gamas de valores.