Как создать собственный градиент с помощью matplotlib

Question

Я уже нормализовал свои данные в цветовую карту, я просто не знаю, как их построить.Вот тип рисунка, который я хочу создать:

Я хочу создать что-то похожее, используя cmap 'bwr', где значение '1'будет сплошным красным, а' 0 'будет сплошным синим.Я использовал это, чтобы нормализовать свои данные и отобразить их в цветах:

norm = Normalize(vmin = min(data), vmax = max(data), clip = True)
mapper = cm.ScalarMappable(norm = norm, cmap = plt.get_cmap('bwr'))

, чтобы data[0] был крайним левым на рисунке, а последнее значение в списке было бы крайним правым.

Answer 1

Если я правильно понимаю, вы хотите создать изображение.Это можно сделать через imshow.

import numpy as np
import matplotlib.pyplot as plt

# some data
data = np.atleast_2d(np.sin(np.sqrt(np.linspace(1,250,250)))*34)
# plot image
plt.imshow(data, aspect="auto", cmap="bwr")
plt.show()