Создание CMap, где цвета связаны с другой переменной

Question

У меня есть разброс график в следующем формате:

Каждая точка на этой диаграмме представляет собой галактики со свойством, известным как скорость образования звезд , Я хочу предписать цветовую карту, зависящую от этой переменной, то есть для более высоких SFR цвет становится более синим, а для более низких SFR цвет становится более красным. (Игнорируйте часть бинограммы с гистограммой для этого упражнения.)

Как бы это достичь? Должен ли я создать и изменить свой собственный cmap?

Answer 1

Вам просто нужен график рассеяния точек с координатами x и y и цвет, представляющий третью переменную?

это то, что разброс для, просто используйте:

import matplotlib.pyplot as plt 

plt.scatter(x, y, c=z, cmap='jet') 

вы можете дать ему любую другую цветовую палитру, все возможности приведены здесь: http://matplotlib.org/examples/color/colormaps_reference.html

здесь небольшой пример:

import matplotlib.pyplot as plt 
import numpy as np 
x = numpy.random.normal(0, 2, 100) 
y = numpy.random.normal(0, 2, 100) 
r = np.sqrt(x**2 + y**2) 
plt.scatter(x, y, c=r, cmap='jet') 

это даст вам 100 2d-гауссовских распределенных точек с цветами в зависимости от расстояния до (0,0)