Mostrar barra de cores no Matplotlib

Neste tutorial, discutiremos como mostrar uma barra de cores em uma figura matplotlib em Python.

Para criar uma barra de cores, temos que usar a função matplotlib.pyplot.colorbar().

O código a seguir mostra um exemplo simples disso.

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

s = plt.scatter(s_x, s_y, c=s_x, cmap="viridis")

c = plt.colorbar()

No exemplo acima, criamos uma barra de cores simples colocada fora do gráfico. Especificamos o mapa de cores usando o parâmetro cmap.

Também podemos especificar os eixos nos quais desejamos mostrar a barra de cores. Se desejarmos, podemos adicioná-lo sobre os eixos do gráfico.

Por exemplo,

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

fig, ax = plt.subplots()
cax = fig.add_axes([0.27, 0.8, 0.5, 0.05])

im = ax.scatter(s_x, s_y, c=s_x, cmap="viridis")
fig.colorbar(im, cax=cax, orientation="horizontal")

No exemplo acima, usamos a função subplots() para obter os objetos de figura e eixos e usá-los para criar os eixos para a barra de cores. Especificamos isso usando o parâmetro cax na função colorbar().

Além disso, observe o uso do parâmetro orientation que alterou a orientação da barra de cores final. Além disso, podemos usar diferentes parâmetros para controlar a forma e o tamanho da barra de cores. Por exemplo, shrink pode reduzir o tamanho da barra de cores por uma pequena margem, aspect, que é a proporção dos lados da barra, e há muitos mais.