Как привязать цветовую карту к массиву?
Я хотел бы получить массив цветов, начиная со значений массива.
Например:
a = [4,3,2,5,6,20,1,34]
Я ожидаю, что у меня будет новый массив с цветом, похожим на viridis cmap из matplotlib, где маленькие числа связаны (тот же индекс) с темными цветами, а большие числа со светлым цветом.
Особых отношений нет, меня интересуют только светлые цвета для более высоких чисел в массиве. Так же, как тепловая карта.
Ответы 2
Вы можете использовать matplotlib.pyplot.Normalize, чтобы данные нормализовались к интервалу [0-1] при подаче на функцию графика.
Вот пример того, как это будет выглядеть, используя нормализованные диапазоны
a = [4,3,2,5,6,20,1,34]
# An example colormap
colormap = plt.cm.cool
# Normalize using the min and max values in a
normalize = plt.Normalize(vmin=min(a), vmax=max(a))
# scatter plot with the parameters defined above
plt.scatter(range(len(a)), y=a, c=a, cmap=colormap, marker='o')
Обратите внимание, что вам придется использовать a в качестве аргумента c в matplotlib.pyplot.scatter, который, как указано в документах, принимает:
Color, sequence, or sequence of color, optional
Таким образом, вы будете использовать массив a для выбора значений из палитры, которые, в свою очередь, будут нормализованы значениями ,ax и min в a.
Спасибо ! Мне нужно было именно то, что возвращает plt.scatter.
Вы можете просто создать массив с цветами RGB следующим образом:
colors = ["rgb(0, 0, 0)"]
И каждое возможное число может иметь запись в массиве. Таким образом, число ноль в этом примере вернет rgb для черного цвета.
как эти числа относятся к цветам?