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:

Plotar matriz 2D Matplotlib usando imshow

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:

Definir cmap Plot Plot 2D array Matplotlib utilizando o imshow

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 na matriz 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çado 2D Matplotlib utilizando pcolormesh

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.