Criando um mapa de cores personalizado e incorporando uma escala de cores

Para criar seu próprio mapa de cores, uma abordagem é utilizar a função LinearSegmentedColormap do módulo matplotlib.colors. Essa abordagem é mais simples e produz uma escala de cores contínua.

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors

# Generate random data points
x, y, c = zip(*np.random.rand(30, 3) * 4 - 2)

# Define lower and upper bounds for normalization
norm = plt.Normalize(-2, 2)

# Create a list of tuples representing the values and corresponding colors
tuples = [(norm(-2.), 'red'), (norm(-1.), 'violet'), (norm(2.), 'blue')]

# Generate the colormap from the list of tuples
cmap = matplotlib.colors.LinearSegmentedColormap.from_list('', tuples)

# Plot the data points using the custom colormap
plt.scatter(x, y, c=c, cmap=cmap, norm=norm)

# Add a color scale to the plot
plt.colorbar()
plt.show()

Este trecho de código cria com sucesso um mapa de cores com uma transição suave do vermelho para o violeta e para o azul, variando de -2 a 2. A escala de cores também é incorporada à direita do gráfico, permitindo para fácil interpretação de cores.