Создание пользовательской цветовой карты и включение цветовой шкалы

Чтобы создать собственную цветовую карту, одним из подходов является использование функции LinearSegmentedColormap из модуля matplotlib.colors. Этот подход проще и создает непрерывную цветовую шкалу.

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

Этот фрагмент кода успешно создает цветовую карту с плавным переходом от красного к фиолетовому и синему в диапазоне от -2 до 2. Цветовая шкала также включена в правую часть графика, что позволяет для облегчения интерпретации цвета.