Как заполнить области цветным градиентом в matplotlib?

Question

У меня есть модель, которая делит точки на плоскости XY на 2 части: она возвращает диапазон значений (0, -1) для каждой входной пары XY выше оранжевой и ниже синей линии, между линиями, которые она возвращает (0, +1). Как я могу заполнить области цветным градиентом в соответствии с этими значениями? Это должно быть очень распространенным делом, но мне не удалось найти ничего более полезного, чем функция fill_between.

Answer 1

Вы можете создать 2D-массив в зависимости от значений вашей модели (с именем z в приведенном ниже коде). А затем раскрасьте пиксели, используя это значение z, убедитесь, что симметрично установлены vmin и vmax, чтобы центральный цвет отмечал ноль вашего z. Если цвета слишком яркие, соответствующая альфа может помочь смягчить их.

from matplotlib import pyplot as plt
import numpy as np

N = 20
red_dots_x = np.concatenate([np.random.normal(0.5, 0.1, N), np.random.normal(-0.5, 0.1, N)])
red_dots_y = np.concatenate([np.random.normal(-0.5, 0.1, N), np.random.normal(0.5, 0.1, N)])
blue_dots_x = np.concatenate([np.random.normal(0.5, 0.1, N), np.random.normal(-0.5, 0.1, N)])
blue_dots_y = np.concatenate([np.random.normal(0.5, 0.1, N), np.random.normal(-0.5, 0.1, N)])

plt.plot(red_dots_x, red_dots_y, 'ro', ls='')
plt.plot(blue_dots_x, blue_dots_y, 'b*', ls='')

x0, x1 = plt.xlim()
y0, y1 = plt.ylim()
x, y = np.meshgrid(np.linspace(x0, x1, 10), np.linspace(y0, y1, 10))
z = abs(x - y) - 0.6
plt.imshow(z, cmap='bwr', vmin=-1, vmax=1, interpolation='bilinear', alpha=0.4,
           extent=[x0, x1, y0, y1], origin='lower')

plt.show()