HOWTO · Matplotlib
Mostrar barra de cores no Matplotlib
Este tutorial demonstra como mostrar uma barra de cores em figuras matplotlib em python
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.
